Upload app.py with huggingface_hub

app.py

@@ -8,6 +8,7 @@ import google.generativeai as genai
 import yaml
 from inference import InferenceManager
 from chat_log import ChatLogger
+from dotenv import load_dotenv
 
 # Configurar logging
 logging.basicConfig(level=logging.DEBUG)
@@ -16,6 +17,10 @@ logger = logging.getLogger(__name__)
 app = Flask(__name__)
 CORS(app)
 
+# Cargar variables de entorno
+load_dotenv()
+GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
+
 # Cargar configuración
 with open('config.yaml', 'r') as file:
     config = yaml.safe_load(file)
@@ -25,10 +30,10 @@ try:
     logger.info("Iniciando sistema...")
     
     logger.info("Configurando API de Google...")
-    genai.configure(api_key=config.get('GOOGLE_API_KEY', ''))
+    genai.configure(api_key=GOOGLE_API_KEY)
     
     logger.info("Inicializando sistema de inferencia...")
-    inference = InferenceManager(config)
+    inference = InferenceManager()
     
     logger.info("Inicializando registro de chat...")
     chat_logger = ChatLogger()
@@ -45,6 +50,9 @@ except Exception as e:
 TEMP_AUDIO_DIR = Path("static/temp_audio")
 TEMP_AUDIO_DIR.mkdir(parents=True, exist_ok=True)
 
+# Configurar el modelo
+model = genai.GenerativeModel('gemini-pro')
+
 @app.after_request
 def after_request(response):
     # Agregar headers necesarios para ngrok
@@ -137,69 +145,40 @@ def send_static(path):
 
 @app.route('/')
 def home():
-    # Agregar mensajes iniciales
-    initial_messages = [{
-        'role': 'assistant',
-        'content': 'Sistema iniciado. Todos los modelos cargados correctamente.'
-    }]
-    return render_template('chat.html', messages=initial_messages)
+    return render_template('chat.html')
 
 @app.route('/chat', methods=['POST'])
 def chat():
     try:
-        message = request.json.get('message')
-        mode = request.json.get('mode', 'seguros')  # Modo por defecto
-        
-        if not message:
-            return jsonify({'error': 'No message provided'}), 400
-            
-        # Registrar mensaje del usuario
-        chat_logger.log_message(message, is_user=True)
+        data = request.json
+        message = data.get('message', '')
+        mode = data.get('mode', 'seguros')
         
-        # Obtener respuesta usando el sistema de inferencia
-        response = inference.get_response(message, mode)
-        
-        if response:
-            # Registrar respuesta del bot
-            chat_logger.log_message(response, is_user=False)
-            
-            return jsonify({
-                'response': response
-            })
-        
-        return jsonify({'error': 'No response generated'}), 500
+        # Actualizar modo si es necesario
+        if mode != inference.current_mode:
+            inference.change_mode(mode)
         
+        response = inference.get_response(message)
+        return jsonify({'response': response})
     except Exception as e:
-        logger.error(f"Error in chat endpoint: {e}", exc_info=True)
         return jsonify({'error': str(e)}), 500
 
 @app.route('/change_mode', methods=['POST'])
 def change_mode():
     try:
         mode = request.json.get('mode')
-        if mode not in ['seguros', 'creditos', 'cobranza']:
-            return jsonify({'error': 'Invalid mode'}), 400
-            
-        # Actualizar modo en el sistema de inferencia
-        inference.mode = mode
-        
-        return jsonify({'success': True})
+        success = inference.change_mode(mode)
+        return jsonify({'success': success})
     except Exception as e:
-        logger.error(f"Error changing mode: {e}")
         return jsonify({'error': str(e)}), 500
 
 @app.route('/change_model', methods=['POST'])
 def change_model():
     try:
         model = request.json.get('model')
-        if model not in ['Gemini 8b', 'Mixtral 7b']:
-            return jsonify({'error': 'Invalid model'}), 400
-            
-        inference.current_model = model
-        
-        return jsonify({'success': True})
+        success = inference.change_model(model)
+        return jsonify({'success': success})
     except Exception as e:
-        logger.error(f"Error changing model: {e}")
         return jsonify({'error': str(e)}), 500
 
 @app.route('/change_tts', methods=['POST'])