Update app.py

app.py

@@ -10,13 +10,18 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 # Función para obtener la respuesta del modelo Gemini
-def get_gemini_response(input_prompt, genre, length, language, mood, target_audience, cta_type):
+# Remove language from function parameters
+def get_gemini_response(input_prompt, genre, length, mood, target_audience, cta_type):
     if not input_prompt:
         return "Por favor, escribe un mensaje para generar contenido."
     
+    # Detect language from input
+    language = "Español" if any(c in input_prompt.lower() for c in "áéíóúñ¿¡") else "English"
+    
     model = genai.GenerativeModel('gemini-2.0-flash')
     full_prompt = f"""
-    You are a creative storyteller. Write a {mood} {genre} in {language} about "{input_prompt}" using everyday words and natural conversation. 
+    You are a creative storyteller. Write a {mood} {genre} about "{input_prompt}" using everyday words and natural conversation. 
+    Write in the same language as the input prompt ({language}).
     Use exactly {length} words.
 
     Target Audience: {target_audience}
@@ -109,13 +114,12 @@ with col1:
                           max_value=150, 
                           value=125,
                           step=5)
-        language = st.selectbox("Idioma del texto:", ["Español", "Inglés"])
         mood = st.selectbox("Estado de ánimo:", ["Emocional", "Triste", "Feliz", "Horror", "Comedia", "Romántico"])
     generate_button = st.button("Generar historia")
 with col2:
     if generate_button:
         response = get_gemini_response(
-            input_prompt, genre, length, language,
+            input_prompt, genre, length,
             mood, target_audience, cta_type
         )
         st.subheader("Contenido generado:")