Update app.py

app.py

@@ -1,17 +1,158 @@
 import gradio as gr
 import os
+import re
+import json
+import requests
+from pypdf import PdfReader
 
-def check_api_key():
-    api_key = os.getenv("DEEPSEEK_API_KEY")
-    if api_key:
-        return f"✅ ¡API Key encontrada! (primeros 10 caracteres: {api_key[:10]}...)"
-    else:
-        return "❌ ERROR: DEEPSEEK_API_KEY no está configurada. Revisa Settings > Variables de entorno."
+# ✅ Obtener API Key (con fallback para diagnóstico)
+DEEPSEEK_API_KEY = os.getenv("DEEPSEEK_API_KEY")
+DEEPSEEK_API_URL = "https://api.deepseek.com/v1/chat/completions"
+MODEL = "deepseek-chat"
+
+def extract_text_from_pdf(pdf_file):
+    try:
+        reader = PdfReader(pdf_file)
+        text = ""
+        for page in reader.pages:
+            extracted = page.extract_text()
+            if extracted:
+                text += extracted + "\n"
+        return text[:5000]
+    except Exception as e:
+        return f"Error al leer PDF: {str(e)}"
+
+def generate_smart_objective(objective, age, duration):
+    age_group = 'preescolar' if age < 36 else 'escolar' if age < 144 else 'adolescente/adulto'
+    time_frame = 'corto plazo' if duration < 30 else 'mediano plazo' if duration < 60 else 'largo plazo'
+    return f"El paciente {age_group} logrará {objective} con un 80% de precisión durante {duration} minutos."
+
+def call_deepseek_api(prompt):
+    if not DEEPSEEK_API_KEY:
+        raise Exception("❌ DEEPSEEK_API_KEY no está configurada. Revisa Settings > Variables de entorno.")
+    
+    headers = {
+        "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
+        "Content-Type": "application/json"
+    }
+    
+    data = {
+        "model": MODEL,
+        "messages": [{"role": "user", "content": prompt}],
+        "temperature": 0.3,
+        "max_tokens": 4096
+    }
+    
+    try:
+        response = requests.post(DEEPSEEK_API_URL, headers=headers, json=data, timeout=60)
+        response.raise_for_status()
+        result = response.json()
+        return result["choices"][0]["message"]["content"]
+    except Exception as e:
+        raise Exception(f"Error de la IA: {str(e)}")
+
+def generate_activity(user_desc, objective, duration, session_type, is_pediatric, context, pdf_files):
+    if not all([user_desc, objective, duration]):
+        return ("⚠️ Error", "Completa todos los campos.", "", "", "", "", "", "")
+
+    try:
+        age = int(re.search(r'\d+', user_desc).group()) if re.search(r'\d+', user_desc) else 60
+        is_child = age < 144 or is_pediatric
+        dur = int(duration)
+
+        pdf_text = ""
+        if pdf_files:
+            for pdf_file in pdf_files:
+                pdf_text += f"\n--- {pdf_file.name} ---\n"
+                pdf_text += extract_text_from_pdf(pdf_file)
+
+        prompt = f"""
+        Eres un fonoaudiólogo experto. Genera una actividad para:
+        PACIENTE: {user_desc}
+        OBJETIVO: {objective}
+        DURACIÓN: {duration} minutos
+        TIPO DE SESIÓN: {session_type}
+        SESIÓN PEDIÁTRICA: {'Sí' if is_child else 'No'}
+        CONTEXTO: {context or 'Ninguno'}
+        REFERENCIAS: {pdf_text[:1000] if pdf_text else 'Ninguna'}
+
+        RESPONDE EN FORMATO JSON: {{"title": "", "smart_objective": "", "description": "", "materials": "", "procedure": "", "evaluation": "", "adaptations": "", "theoretical_foundation": ""}}
+        """
+
+        content = call_deepseek_api(prompt)
+
+        # Modo fallback: si falla el JSON, devuelve texto plano
+        try:
+            if content.startswith("```json"):
+                content = content[7:]
+            if content.endswith("```"):
+                content = content[:-3]
+            data = json.loads(content)
+        except:
+            return (
+                "Actividad Generada",
+                "Objetivo generado por IA",
+                content,
+                "Materiales generados por IA",
+                "Procedimiento generado por IA",
+                "Evaluación generada por IA",
+                "Adaptaciones generadas por IA",
+                "Fundamentación generada por IA"
+            )
+
+        return (
+            data.get("title", "Actividad Generada"),
+            data.get("smart_objective", generate_smart_objective(objective, age, dur)),
+            data.get("description", "Descripción generada por IA."),
+            data.get("materials", "Materiales generados por IA."),
+            data.get("procedure", "Procedimiento generado por IA."),
+            data.get("evaluation", "Evaluación generada por IA."),
+            data.get("adaptations", "Adaptaciones generadas por IA."),
+            data.get("theoretical_foundation", "Fundamentación generada por IA.")
+        )
+
+    except Exception as e:
+        return (
+            "❌ Error",
+            str(e),
+            "Por favor verifica tu API Key en Settings > Variables de entorno y reinicia el Space.",
+            "Si el problema persiste, contacta al desarrollador.",
+            "",
+            "",
+            "",
+            ""
+        )
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🧪 Diagnóstico de API Key")
-    output = gr.Textbox(label="Estado de la API Key")
-    btn = gr.Button("Verificar API Key")
-    btn.click(fn=check_api_key, inputs=[], outputs=output)
+    gr.Markdown("# 🧠 Generador IA de Actividades Fonoaudiológicas")
+    
+    with gr.Row():
+        with gr.Column():
+            user_desc = gr.Textbox(label="👤 Descripción del usuario", placeholder="Ej: Niño de 48 meses...")
+            objective = gr.Textbox(label="🎯 Objetivo específico")
+            duration = gr.Number(label="⏱️ Duración (minutos)", value=30)
+            session_type = gr.Dropdown(["individual", "grupal", "hogar"], label="👥 Tipo de sesión")
+            is_pediatric = gr.Checkbox(label="🧸 Sesión Pediátrica")
+            context = gr.Textbox(label="📚 Contexto Adicional", lines=3)
+            pdf_files = gr.File(label="📖 Subir PDFs", file_types=[".pdf"], file_count="multiple")
+            btn = gr.Button("✨ Generar Actividad con IA")
+        
+        with gr.Column():
+            title = gr.Textbox(label="Título")
+            smart_obj = gr.Textbox(label="📋 Objetivo SMART")
+            description = gr.Textbox(label="📝 Descripción", lines=3)
+            materials = gr.Textbox(label="🎯 Materiales", lines=3)
+            procedure = gr.Textbox(label="⚡ Procedimiento", lines=3)
+            evaluation = gr.Textbox(label="📊 Evaluación", lines=3)
+            adaptations = gr.Textbox(label="🔧 Adaptaciones", lines=3)
+            theory = gr.Textbox(label="📚 Fundamentación Teórica", lines=3)
+    
+    btn.click(
+        fn=generate_activity,
+        inputs=[user_desc, objective, duration, session_type, is_pediatric, context, pdf_files],
+        outputs=[title, smart_obj, description, materials, procedure, evaluation, adaptations, theory]
+    )
+
+demo.launch()
 
 demo.launch()