Update app.py

app.py

@@ -46,29 +46,13 @@ def extract_file_text(file):
     if not file:
         return ""
     try:
-        # Si Gradio pasa una lista de archivos, tomar el primero
-        if isinstance(file, (list, tuple)):
-            if len(file) == 0:
-                return ""
-            file = file[0]
-
-        # Obtener la ruta del archivo
-        path = getattr(file, "name", None) or (file if isinstance(file, str) else None)
-        if not path:
-            return ""
-
-        # Validar que sea un archivo y no un directorio
-        if os.path.isdir(path):
-            return ""
-        if not os.path.isfile(path):
-            return ""
-
+        path = file.name if hasattr(file, 'name') else file
         with open(path, 'r', encoding='utf-8', errors='ignore') as f:
             return f.read()[:20000]
-    except Exception as e:
-        print(f"[DEBUG] extract_file_text error: {e}")
+    except Exception:
         return ""
 
+
 def get_ext(nombre):
     n = nombre.lower()
     if n.endswith((".html", ".htm")):
@@ -303,143 +287,128 @@ def history_to_messages(norm_history):
 
 
 def run_chat(message, history, user_context, max_tokens, temperature, top_p, token, file_content, file_name, memoria):
-    try:
-        # --- Normalizar history ---
-        norm_history = normalize_history(history)
-
-        # --- Validar mensaje vacío ---
-        if not message or not message.strip():
-            yield [], "", "", "⚠️ Mensaje vacío.", ""
-            return
-
-        # --- SEGURIDAD ---
-        for bloqueo in BLOQUEOS_JAILBREAK:
-            if bloqueo in message.lower():
-                reply = {"role": "assistant", "content": "Soy Aura, asistente especializado exclusivamente en programación."}
-                chatbot_output = norm_history + [{"role": "user", "content": message}, reply]
-                yield chatbot_output, "", "", "⛔ Bloqueado por seguridad.", ""
-                return
+    # --- Normalizar history ---
+    norm_history = normalize_history(history)
 
-        if not HF_TOKEN:
-            reply = {"role": "assistant", "content": "⚠️ Falta HF_TOKEN. Configura tu API Key en Settings → Secrets."}
+    # --- SEGURIDAD ---
+    for bloqueo in BLOQUEOS_JAILBREAK:
+        if bloqueo in message.lower():
+            reply = {"role": "assistant", "content": "Soy Aura, asistente especializado exclusivamente en programación."}
             chatbot_output = norm_history + [{"role": "user", "content": message}, reply]
-            yield chatbot_output, "", "", "Configura HF_TOKEN.", ""
-            return
-
-        client = InferenceClient(token=token or HF_TOKEN, model=MODEL_ID)
-
-        # --- Contexto del archivo ---
-        contexto_archivo = ""
-        errores_iniciales = []
-        if file_content and file_name:
-            errores_iniciales = analizar_todo(file_content, file_name)
-            contexto_archivo = f"\n--- ARCHIVO ({file_name}) ---\n{file_content}\n"
-            if errores_iniciales:
-                contexto_archivo += "\n--- ERRORES ---\n" + "\n".join(errores_iniciales) + "\n"
-
-        final_system = SYSTEM_PROMPT_BASE
-        if user_context and user_context.strip():
-            final_system += f"\n\n--- TU CONTEXTO ---\n{user_context}\n"
-        if memoria and memoria.strip():
-            final_system += f"\n--- MEMORIA ---\n{memoria}\n"
-
-        # --- Construir messages para la API ---
-        messages = [{"role": "system", "content": final_system}]
-        messages.extend(history_to_messages(norm_history))
-        user_content = message + (contexto_archivo if contexto_archivo else "")
-        messages.append({"role": "user", "content": user_content})
-
-        # --- Validar mensajes ---
-        for i, msg in enumerate(messages):
-            if not isinstance(msg, dict) or "role" not in msg or "content" not in msg:
-                messages[i] = {"role": "system", "content": "[mensaje corrupto eliminado]"}
-            msg["content"] = safe_str(msg["content"])
-
-        # --- Streaming ---
-        resp_text = ""
-        try:
-            for chunk in client.chat_completion(
-                messages=messages,
-                max_tokens=max_tokens,
-                stream=True,
-                temperature=temperature,
-                top_p=top_p
-            ):
-                if chunk.choices and chunk.choices[0].delta.content:
-                    resp_text += chunk.choices[0].delta.content
-                    chatbot_so_far = norm_history + [
-                        {"role": "user", "content": message},
-                        {"role": "assistant", "content": resp_text}
-                    ]
-                    yield chatbot_so_far, "", "", "Generando...", ""
-        except Exception as e:
-            error_msg = f"Error IA: {e}"
-            print(f"[DEBUG] Error streaming: {e}")
-            chatbot_output = norm_history + [
-                {"role": "user", "content": message},
-                {"role": "assistant", "content": error_msg}
-            ]
-            yield chatbot_output, "", "", error_msg, ""
+            yield chatbot_output, "", "", "⛔ Bloqueado por seguridad.", ""
             return
 
-        # --- Auto-corrección ---
-        intentos = 0
-        ultima_ejecucion = ""
-        code_blocks = re.findall(r'```(?:python|javascript|js|html|sql)?\n(.*?)\n```', resp_text, re.DOTALL)
-
-        if code_blocks:
-            main_code = max(code_blocks, key=len)
-            detected_ext = get_ext(file_name) if file_name else ("js" if ("function" in main_code or "const " in main_code) else "py")
-            filename_for_exec = f"test.{detected_ext}"
-
-            while intentos < MAX_AUTOFIX:
-                exito, resultado_exec = ejecutar_codigo_sandbox(main_code, filename_for_exec)
-                ultima_ejecucion = resultado_exec
-                if exito:
-                    break
-                intentos += 1
+    if not HF_TOKEN:
+        reply = {"role": "assistant", "content": "⚠️ Falta HF_TOKEN. Configura tu API Key en Settings → Secrets."}
+        chatbot_output = norm_history + [{"role": "user", "content": message}, reply]
+        yield chatbot_output, "", "", "Configura HF_TOKEN.", ""
+        return
+
+    client = InferenceClient(token=token or HF_TOKEN, model=MODEL_ID)
+
+    # --- Contexto del archivo ---
+    contexto_archivo = ""
+    errores_iniciales = []
+    if file_content and file_name:
+        errores_iniciales = analizar_todo(file_content, file_name)
+        contexto_archivo = f"\n--- ARCHIVO ({file_name}) ---\n{file_content}\n"
+        if errores_iniciales:
+            contexto_archivo += "\n--- ERRORES ---\n" + "\n".join(errores_iniciales) + "\n"
+
+    final_system = SYSTEM_PROMPT_BASE
+    if user_context and user_context.strip():
+        final_system += f"\n\n--- TU CONTEXTO ---\n{user_context}\n"
+    if memoria and memoria.strip():
+        final_system += f"\n--- MEMORIA ---\n{memoria}\n"
+
+    # --- Construir messages para la API ---
+    messages = [{"role": "system", "content": final_system}]
+    messages.extend(history_to_messages(norm_history))
+    user_content = message + (contexto_archivo if contexto_archivo else "")
+    messages.append({"role": "user", "content": user_content})
+
+    # --- Validar mensajes ---
+    for i, msg in enumerate(messages):
+        if not isinstance(msg, dict) or "role" not in msg or "content" not in msg:
+            messages[i] = {"role": "system", "content": "[mensaje corrupto eliminado]"}
+        msg["content"] = safe_str(msg["content"])
+
+    # --- Streaming ---
+    resp_text = ""
+    try:
+        for chunk in client.chat_completion(
+            messages=messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p
+        ):
+            if chunk.choices and chunk.choices[0].delta.content:
+                resp_text += chunk.choices[0].delta.content
                 chatbot_so_far = norm_history + [
                     {"role": "user", "content": message},
-                    {"role": "assistant", "content": f"{resp_text}\n\n🔄 *Auto-corrigiendo ({intentos}/{MAX_AUTOFIX})...*"}
+                    {"role": "assistant", "content": resp_text}
                 ]
-                yield chatbot_so_far, "", "", ultima_ejecucion, ""
-                try:
-                    new_resp = ""
-                    for chunk in client.chat_completion(
-                        messages=[
-                            {"role": "system", "content": final_system},
-                            {"role": "user", "content": f"CORRIGE ESTE ERROR:\n{resultado_exec}\nDevuelve SOLO el código corregido en ```{detected_ext}."}
-                        ],
-                        max_tokens=max_tokens,
-                        stream=True,
-                        temperature=0.2
-                    ):
-                        if chunk.choices and chunk.choices[0].delta.content:
-                            new_resp += chunk.choices[0].delta.content
-                    new_blocks = re.findall(r'```(?:python|javascript|js|html|sql)?\n(.*?)\n```', new_resp, re.DOTALL)
-                    if new_blocks:
-                        main_code = max(new_blocks, key=len)
-                        resp_text += f"\n\n--- CORRECCIÓN {intentos} ---\n{new_resp}"
-                    else:
-                        break
-                except Exception as e:
-                    print(f"\n[DEBUG] Error auto-corrección {intentos}: {e}")
-                    break
-
-        # --- Output final ---
+                yield chatbot_so_far, "", "", "Generando...", ""
+    except Exception as e:
+        error_msg = f"Error IA: {e}"
         chatbot_output = norm_history + [
             {"role": "user", "content": message},
-            {"role": "assistant", "content": resp_text}
+            {"role": "assistant", "content": error_msg}
         ]
-        yield chatbot_output, resp_text, resp_text, ultima_ejecucion, memoria
-
-    except Exception as e:
-        # CATCH-ALL: Si algo explota antes del primer yield, esto lo salva
-        import traceback
-        error_trace = traceback.format_exc()
-        print(f"[DEBUG] CRASH EN RUN_CHAT: {error_trace}")
-        yield [], "", "", f"💥 Error interno: {str(e)}", ""
-
+        yield chatbot_output, "", "", error_msg, ""
+        return
+
+    # --- Auto-corrección ---
+    intentos = 0
+    ultima_ejecucion = ""
+    code_blocks = re.findall(r'```(?:python|javascript|js|html|sql)?\n(.*?)\n```', resp_text, re.DOTALL)
+
+    if code_blocks:
+        main_code = max(code_blocks, key=len)
+        detected_ext = get_ext(file_name) if file_name else ("js" if ("function" in main_code or "const " in main_code) else "py")
+        filename_for_exec = f"test.{detected_ext}"
+
+        while intentos < MAX_AUTOFIX:
+            exito, resultado_exec = ejecutar_codigo_sandbox(main_code, filename_for_exec)
+            ultima_ejecucion = resultado_exec
+            if exito:
+                break
+            intentos += 1
+            chatbot_so_far = norm_history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": f"{resp_text}\n\n🔄 *Auto-corrigiendo ({intentos}/{MAX_AUTOFIX})...*"}
+            ]
+            yield chatbot_so_far, "", "", ultima_ejecucion, ""
+            try:
+                new_resp = ""
+                for chunk in client.chat_completion(
+                    messages=[
+                        {"role": "system", "content": final_system},
+                        {"role": "user", "content": f"CORRIGE ESTE ERROR:\n{resultado_exec}\nDevuelve SOLO el código corregido en ```{detected_ext}."}
+                    ],
+                    max_tokens=max_tokens,
+                    stream=True,
+                    temperature=0.2
+                ):
+                    if chunk.choices and chunk.choices[0].delta.content:
+                        new_resp += chunk.choices[0].delta.content
+                new_blocks = re.findall(r'```(?:python|javascript|js|html|sql)?\n(.*?)\n```', new_resp, re.DOTALL)
+                if new_blocks:
+                    main_code = max(new_blocks, key=len)
+                    resp_text += f"\n\n--- CORRECCIÓN {intentos} ---\n{new_resp}"
+                else:
+                    break
+            except Exception as e:
+                print(f"\n[DEBUG] Error auto-corrección {intentos}: {e}")
+                break
+
+    # --- Output final ---
+    chatbot_output = norm_history + [
+        {"role": "user", "content": message},
+        {"role": "assistant", "content": resp_text}
+    ]
+    yield chatbot_output, resp_text, resp_text, ultima_ejecucion, memoria
 # ============================================
 # INTERFAZ GRADIO
 # ============================================