Upload app.py with huggingface_hub

app.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 import os
 import json
+import tempfile
 import requests
 import gradio as gr
 
@@ -9,57 +10,88 @@ import gradio as gr
 # ----------------------------------------------------------------------
 GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
 GROQ_ENDPOINT = "https://api.groq.com/openai/v1/chat/completions"
-
-# Default model can be overridden by the user via the dropdown
-DEFAULT_MODEL = os.environ.get("MODEL_NAME", "llama-3.1-8b-instant")
-AVAILABLE_MODELS = [
-    "llama-3.3-70b-versatile",
-    "llama-3.1-8b-instant",
-    "qwen/qwen3-32b"
-]
+MODEL_NAME = os.environ.get("MODEL_NAME", "llama-3.1-8b-instant")
 
 # ----------------------------------------------------------------------
 # CSS (will be passed to demo.launch)
 # ----------------------------------------------------------------------
 CSS = """
-#chatbot .message {
-    font-family: Arial, Helvetica, sans-serif;
+#business-input textarea {
+    border: 1px solid #CCCCCC;
+}
+#business-input textarea:focus {
+    border-color: #1E90FF;
 }
-#submit-btn {
+#generate-btn {
     background-color: #1E90FF;
     color: white;
-    border: none;
-    padding: 0.5rem 1rem;
-    font-size: 1rem;
-    cursor: pointer;
 }
-#submit-btn:hover {
+#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;
+}
 """
 
 # ----------------------------------------------------------------------
-# Helper to call Groq API
+# Helper functions
 # ----------------------------------------------------------------------
-def call_groq(messages, model):
+def call_groq(business_name: str) -> dict:
     """
-    Sends a chat completion request to Groq and returns the assistant's reply.
+    Calls the Groq API and returns a dict with keys:
+    problems, solutions, explanation, eos_prompt.
     """
     if not GROQ_API_KEY:
-        raise RuntimeError("GROQ_API_KEY not set in environment.")
+        raise RuntimeError("GROQ_API_KEY not set in environment")
+
+    user_prompt = f"""Eres un analista de negocios experto. Dado el nombre del negocio "{business_name}", haz una lista de los 5 problemas clasicos principales, propone una solucion para cada uno, ofrece una explicacion detallada combinada y, por ultimo, genera un prompt de EOS listo para usar. 
+
+Toda la respuesta debe estar redactada en español, pero estructurada estrictamente en el siguiente formato JSON (manteniendo las claves en ingles para la lectura del sistema):
+{{
+  "problems": ["...", "...", "..."],
+  "solutions": ["...", "...", "..."],
+  "explanation": "...",
+  "eos_prompt": "Titulo: ...\\nContexto: ...\\nInstrucciones: ..."
+}}
+Devuelve unica y exclusivamente el objeto JSON, sin texto introductorio ni conclusiones."""
+    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"
+        "Content-Type": "application/json",
     }
-    payload = {
-        "model": model,
-        "messages": messages,
-        "temperature": 0.7
-    }
-    response = requests.post(GROQ_ENDPOINT, headers=headers, json=payload, timeout=30)
+
+    response = requests.post(
+        GROQ_ENDPOINT,
+        headers=headers,
+        json=payload,
+        timeout=30,
+    )
     response.raise_for_status()
     data = response.json()
-    # Extract the assistant's content
-    content = data["choices"][0]["message"]["content"]
-    # Remove surrounding markdown fences if present
+    content = data["choices"][0]["message"]["content"].strip()
+
+    # Remove possible markdown fences
     if content.startswith("