Update app.py

app.py

@@ -1,20 +1,91 @@
 import os
+import gradio as gr
 import requests
+import json
 
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-url = "https://api.groq.com/openai/v1/chat/completions"
-headers = {
-    "Authorization": f"Bearer {GROQ_API_KEY}",
-    "Content-Type": "application/json"
-}
-body = {
-    "model": "llama-3.1-8b-instant",
-    "messages": [
-        {"role": "user", "content": "Say hello!"}
-    ],
-    "temperature": 0.3,
-    "max_tokens": 50
-}
-response = requests.post(url, headers=headers, json=body)
-print(response.status_code)
-print(response.text)
+GROQ_COMPLETION_URL = "https://api.groq.com/openai/v1/chat/completions"
+GROQ_MODEL = "llama-3.1-8b-instant"
+
+def groq_completion(prompt, system_prompt=None):
+    headers = {
+        "Authorization": f"Bearer {GROQ_API_KEY}",
+        "Content-Type": "application/json"
+    }
+    body = {
+        "model": GROQ_MODEL,
+        "messages": [
+            {"role": "system", "content": system_prompt or "You are a fast and reliable business email classification assistant."},
+            {"role": "user", "content": prompt}
+        ],
+        "temperature": 0.3,
+        "max_tokens": 512
+    }
+    try:
+        response = requests.post(GROQ_COMPLETION_URL, headers=headers, json=body)
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]
+    except Exception as e:
+        print("Groq API error:", e)
+        if 'response' in locals():
+            print("Groq API response text:", response.text)
+        return "Error"
+
+def email_assistant(email_text):
+    # Clear, strict prompt for JSON extraction
+    prompt = (
+        "Analyze the following business email and extract ONLY the following elements as a compact JSON dictionary:\n"
+        "Category (Sales, Support, Spam, Finance, General),\n"
+        "Confidence (percentage),\n"
+        "Priority (High/Normal/Low),\n"
+        "Suggested Action or Forwarding Address,\n"
+        "Draft professional response,\n"
+        "Action items (bulleted list).\n\n"
+        "Return ONLY a compact JSON dictionary, nothing else:\n"
+        '{"Category":"...", "Confidence":"...", "Priority":"...", "Suggested Action":"...", "Draft Response":"...", "Action Items":"..."}\n\n'
+        f"EMAIL:\n{email_text}\n"
+    )
+    raw_result = groq_completion(prompt)
+    try:
+        output = json.loads(raw_result)
+        return [
+            output.get("Category", "Error"),
+            output.get("Confidence", "Error"),
+            output.get("Priority", "Error"),
+            output.get("Suggested Action", "Error"),
+            output.get("Draft Response", "Error"),
+            output.get("Action Items", "Error"),
+        ]
+    except Exception as e:
+        print(f"Parsing or LLM output error: {e}\nRaw output: {raw_result}")
+        # Fail gracefully: show result in all fields for debugging
+        return [raw_result]*6
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Email Classification & Response Assistant (Groq powered)")
+    email_input = gr.TextArea(
+        label="Paste your email here (include Subject, From, Body if possible)",
+        placeholder="Subject: ...\nFrom: ...\nBody: ..."
+    )
+    submit = gr.Button("Analyze Email")
+    output_category = gr.Textbox(label="Category")
+    output_confidence = gr.Textbox(label="Confidence")
+    output_priority = gr.Textbox(label="Priority")
+    output_action = gr.Textbox(label="Suggested Action")
+    output_response = gr.Textbox(label="Draft Response")
+    output_items = gr.Textbox(label="Extracted Action Items")
+    submit.click(
+        email_assistant,
+        inputs=email_input,
+        outputs=[
+            output_category,
+            output_confidence,
+            output_priority,
+            output_action,
+            output_response,
+            output_items,
+        ]
+    )
+
+if __name__ == "__main__":
+    demo.launch()