Upload app.py with huggingface_hub

app.py

@@ -6,25 +6,26 @@ import requests
 import gradio as gr
 
 # ----------------------------------------------------------------------
-# Configuracion de la API
+# Configuration
 # ----------------------------------------------------------------------
 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 para la aplicacion (se pasa a demo.launch)
+# CSS (passed to demo.launch)
 # ----------------------------------------------------------------------
 CSS = """
-#business-input textarea {
-    border: 1px solid #CCCCCC;
-}
-#business-input textarea:focus {
-    border-color: #1E90FF;
+#chatbot .message {
+    font-family: Arial, Helvetica, sans-serif;
 }
 #generate-btn {
     background-color: #1E90FF;
     color: white;
+    border: none;
+    padding: 0.5rem 1rem;
+    font-size: 1rem;
+    cursor: pointer;
 }
 #generate-btn:hover {
     background-color: #1C86EE;
@@ -32,16 +33,27 @@ CSS = """
 """
 
 # ----------------------------------------------------------------------
-# Utilidades
+# Helper to call Groq API
 # ----------------------------------------------------------------------
-def extract_json(text: str) -> dict:
+def call_groq(messages):
     """
-    Extrae el primer bloque JSON valido del texto recibido.
+    Sends a chat completion request to Groq and returns the assistant's reply.
     """
-    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 bloques de codigo markdown si existen
-    json_str = re.sub(r"^
+    if not GROQ_API_KEY:
+        raise RuntimeError("GROQ_API_KEY not set in environment.")
+    headers = {
+        "Authorization": f"Bearer {GROQ_API_KEY}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "model": MODEL_NAME,
+        "messages": messages,
+        "temperature": 0.7
+    }
+    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"]
+    # Some models wrap the answer in markdown code fences; clean them
+    if content.startswith("