Update app.py

app.py

@@ -2,6 +2,7 @@ from dotenv import load_dotenv
 import streamlit as st
 import os
 import google.generativeai as genai
+from langdetect import detect
 
 # Cargar variables de entorno
 load_dotenv()
@@ -15,9 +16,13 @@ def get_gemini_response(input_prompt, genre, length, mood, target_audience, cta_
     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"
-    
+     # Better language detection
+    try:
+        detected_lang = detect(input_prompt)
+        language = "Español" if detected_lang == "es" else "English"
+    except:
+        language = "Español"  # Default to Spanish if detection fails
+        
     model = genai.GenerativeModel('gemini-2.0-flash')
     full_prompt = f"""
     You are a creative storyteller. Write a {mood} {genre} about "{input_prompt}" using everyday words and natural conversation.