Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 import time
 import uuid
 from typing import List, Tuple, Optional, Dict, Union
+from langdetect import detect
 
 import google.generativeai as genai
 import gradio as gr
@@ -58,6 +59,12 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
         chatbot.append((text_prompt, None))
     return "", chatbot
 
+def detect_language(text: str) -> str:
+    try:
+        return detect(text)
+    except:
+        return "en"  # En caso de no poder detectar el idioma, se usa el inglés por defecto
+
 def bot(
     files: Optional[List[str]],
     temperature: float,
@@ -72,12 +79,19 @@ def bot(
 
     # Configurar la API con la clave
     genai.configure(api_key=GOOGLE_API_KEY)
+    
+    # Detectar el idioma del último mensaje del usuario
+    last_user_message = chatbot[-1][0] if chatbot and chatbot[-1][0] else ""
+    detected_language = detect_language(last_user_message)
+
+    # Configurar la generación para que utilice el idioma detectado
     generation_config = genai.types.GenerationConfig(
         temperature=temperature,
         max_output_tokens=max_output_tokens,
         stop_sequences=preprocess_stop_sequences(stop_sequences=stop_sequences),
         top_k=top_k,
-        top_p=top_p
+        top_p=top_p,
+        language=detected_language  # Usar el idioma detectado
     )
 
     text_prompt = [chatbot[-1][0]] if chatbot and chatbot[-1][0] and isinstance(chatbot[-1][0], str) else []