Upload app.py with huggingface_hub

app.py

@@ -1,19 +1,20 @@
 # -*- coding: utf-8 -*-
 import os
 import json
-import requests
+import re
 import tempfile
+import requests
 import gradio as gr
 
 # ----------------------------------------------------------------------
-# Configuracion
+# Configuracion de la API
 # ----------------------------------------------------------------------
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 GROQ_ENDPOINT = "https://api.groq.com/openai/v1/chat/completions"
 MODEL_NAME = os.environ.get("MODEL_NAME", "llama-3.1-8b-instant")
 
 # ----------------------------------------------------------------------
-# CSS (se pasa a demo.launch)
+# CSS para la aplicacion (se pasa a demo.launch)
 # ----------------------------------------------------------------------
 CSS = """
 #business-input textarea {
@@ -29,68 +30,20 @@ CSS = """
 #generate-btn:hover {
     background-color: #1C86EE;
 }
-#problems-card, #solutions-card, #explanation-card, #prompt-card {
-    padding: 15px;
-    border-radius: 8px;
-    margin-top: 10px;
-}
-#problems-card {
-    background-color: #F0F8FF;
-}
-#solutions-card {
-    background-color: #E6F7FF;
-}
-#explanation-card {
-    background-color: #FFFFFF;
-}
-#prompt-card {
-    background-color: #F5F5F5;
-}
-#download-btn {
-    margin-top: 10px;
-}
 """
 
 # ----------------------------------------------------------------------
-# Funciones auxiliares
+# Utilidades
 # ----------------------------------------------------------------------
-def call_groq(business_name: str) -> dict:
+def extract_json(text: str) -> dict:
     """
-    Llama a la API de Groq y devuelve un dict con:
-    problems, solutions, explanation, eos_prompt.
+    Extrae el primer bloque JSON valido del texto recibido.
     """
-    if not GROQ_API_KEY:
-        raise RuntimeError("GROQ_API_KEY no esta definida en el entorno")
-
-    user_prompt = f"""Eres un analista de negocios experto. Dado el nombre del negocio "{business_name}", enumera los 5 problemas clasicos mas comunes, propone una solucion para cada uno, brinda una explicacion combinada detallada, y finalmente genera un prompt EOS listo para usar en el siguiente formato JSON:
-{{
-  "problems": ["..."],
-  "solutions": ["..."],
-  "explanation": "...",
-  "eos_prompt": "Title: ...\\nContext: ...\\nInstructions: ..."
-}}
-Devuelve SOLO el objeto JSON, sin texto adicional."""
-
-    payload = {
-        "model": MODEL_NAME,
-        "messages": [{"role": "user", "content": user_prompt}],
-        "temperature": 0.7,
-        "max_tokens": 1024,
-    }
-    headers = {
-        "Authorization": f"Bearer {GROQ_API_KEY}",
-        "Content-Type": "application/json",
-    }
-
-    response = requests.post(
-        GROQ_ENDPOINT,
-        headers=headers,
-        json=payload,
-        timeout=30,
-    )
-    response.raise_for_status()
-    data = response.json()
-    content = data["choices"][0]["message"]["content"].strip()
-
-    # El modelo a veces envuelve el JSON en bloques de codigo
-    if content.startswith("
+    # Busca el primer { y el ultimo }
+    start = text.find("{")
+    end = text.rfind("}")
+    if start == -1 or end == -1:
+        raise ValueError("No JSON found in response")
+    json_str = text[start:end + 1]
+    # Elimina posibles bloques de codigo markdown
+    json_str = re.sub(r"^