Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,210 @@
+# -*- coding: utf-8 -*-
+import os
+import json
+import requests
+import tempfile
+import gradio as gr
+
+# ----------------------------------------------------------------------
+# Configuration
+# ----------------------------------------------------------------------
+GROQ_API_KEY = os.environ.get("GROQ_API_KEY")
+GROQ_ENDPOINT = "https://api.groq.com/v1/chat/completions"
+MODEL_NAME = "llama-3.1-8b-instant"  # default model, can be overridden via env if needed
+
+# ----------------------------------------------------------------------
+# Helper functions
+# ----------------------------------------------------------------------
+def call_groq(business_name: str) -> dict:
+    """
+    Calls Groq API with a prompt that asks for a JSON containing:
+    problems, solutions, explanation, eos_prompt.
+    Returns a dict with those keys or raises an Exception.
+    """
+    if not GROQ_API_KEY:
+        raise RuntimeError("GROQ_API_KEY not set in environment")
+
+    user_prompt = f"""You are an expert business analyst. Given the business name "{business_name}", list the top 5 classic problems, propose a solution for each, provide a detailed combined explanation, and finally output a ready-to-use EOS prompt in the following JSON format:
+{{
+  "problems": ["..."],
+  "solutions": ["..."],
+  "explanation": "...",
+  "eos_prompt": "Title: ...\\nContext: ...\\nInstructions: ..."
+}}
+Only return the JSON object, no additional text."""
+    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=10,
+    )
+    response.raise_for_status()
+    data = response.json()
+    # Expected structure: {"choices": [{"message": {"content": "..."} }]}
+    content = data["choices"][0]["message"]["content"]
+    # The model should return a JSON string
+    return json.loads(content)
+
+
+def format_markdown_list(items):
+    """Convert a list of strings into a numbered markdown list."""
+    return "\n".join([f"{i+1}. {item}" for i, item in enumerate(items)])
+
+
+def generate(business_name):
+    """
+    Main function called when the user clicks "Generar".
+    Returns:
+        problems_md, solutions_md, explanation_md, eos_prompt_code, json_path
+    """
+    if not business_name or not business_name.strip():
+        error_md = "### ❗ Error: El nombre del negocio no puede estar vacío."
+        return error_md, error_md, error_md, error_md, None
+
+    try:
+        result = call_groq(business_name.strip())
+        problems = result.get("problems", [])
+        solutions = result.get("solutions", [])
+        explanation = result.get("explanation", "")
+        eos_prompt = result.get("eos_prompt", "")
+
+        # Basic validation
+        if not (problems and solutions and explanation and eos_prompt):
+            raise ValueError("Respuesta incompleta del modelo.")
+
+        problems_md = f"### Problemas\n{format_markdown_list(problems)}"
+        solutions_md = f"### Soluciones\n{format_markdown_list(solutions)}"
+        explanation_md = f"### Explicación\n{explanation}"
+        eos_prompt_code = eos_prompt  # will be shown in a gr.Code component
+
+        # Create temporary JSON file for download
+        export_dict = {
+            "business_name": business_name,
+            "problems": problems,
+            "solutions": solutions,
+            "explanation": explanation,
+            "eos_prompt": eos_prompt,
+        }
+        tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".json", mode="w", encoding="utf-8")
+        json.dump(export_dict, tmp_file, ensure_ascii=False, indent=2)
+        tmp_file.close()
+        json_path = tmp_file.name
+
+        return problems_md, solutions_md, explanation_md, eos_prompt_code, json_path
+
+    except requests.exceptions.RequestException as e:
+        err_md = f"### ❗ Error al conectar con Groq: {str(e)}"
+        return err_md, err_md, err_md, err_md, None
+    except (json.JSONDecodeError, ValueError) as e:
+        err_md = f"### ❗ Error al procesar la respuesta: {str(e)}"
+        return err_md, err_md, err_md, err_md, None
+    except Exception as e:
+        err_md = f"### ❗ Error inesperado: {str(e)}"
+        return err_md, err_md, err_md, err_md, None
+
+
+# ----------------------------------------------------------------------
+# Gradio Interface
+# ----------------------------------------------------------------------
+with gr.Blocks() as demo:
+    # Header
+    gr.Markdown(
+        "<h1 style='text-align:center; font-family:Roboto; font-weight:bold; font-size:24px;'>Business Problem Solver</h1>"
+    )
+
+    # Input form
+    with gr.Row():
+        business_input = gr.Textbox(
+            label="Nombre del negocio",
+            placeholder="Introduce el nombre del negocio",
+            lines=1,
+            max_lines=1,
+            elem_id="business-input",
+        )
+        generate_btn = gr.Button(
+            "Generar",
+            variant="primary",
+            elem_id="generate-btn",
+        )
+
+    # Output cards
+    with gr.Column():
+        problems_md = gr.Markdown(elem_id="problems-card")
+        solutions_md = gr.Markdown(elem_id="solutions-card")
+        explanation_md = gr.Markdown(elem_id="explanation-card")
+        eos_prompt_code = gr.Code(
+            label="Prompt EOS",
+            language="text",
+            show_copy_button=True,
+            elem_id="prompt-card",
+        )
+        download_btn = gr.DownloadButton(
+            label="Descargar JSON",
+            value=None,
+            elem_id="download-btn",
+        )
+
+    # Footer
+    gr.Markdown(
+        "<small>Más información sobre la API de Groq: <a href='https://groq.com/docs' target='_blank'>Documentación Groq</a></small>",
+        elem_id="footer",
+    )
+
+    # Interaction
+    generate_btn.click(
+        fn=generate,
+        inputs=[business_input],
+        outputs=[problems_md, solutions_md, explanation_md, eos_prompt_code, download_btn],
+    )
+
+# ----------------------------------------------------------------------
+# Custom CSS
+# ----------------------------------------------------------------------
+css = """
+#business-input textarea {
+    border: 1px solid #CCCCCC;
+}
+#business-input textarea:focus {
+    border-color: #1E90FF;
+}
+#generate-btn {
+    background-color: #1E90FF;
+    color: white;
+}
+#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;
+}
+"""
+
+demo.launch(css=css)