Update app.py

app.py

@@ -1,178 +1,101 @@
 import os
-from typing import List, Optional, Tuple, Any
-
+import requests
 import gradio as gr
-from groq import Groq
-from pdf2image import convert_from_path
-from PIL import Image
-
-from llava13b_runtime import analyze_image
 
+# -----------------------------
+# ENVIRONMENT
+# -----------------------------
 GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 if not GROQ_API_KEY:
-    raise ValueError("GROQ_API_KEY environment variable is not set.")
+    raise ValueError("GROQ_API_KEY is not set.")
 
-groq_client = Groq(api_key=GROQ_API_KEY)
+GROQ_URL = "https://api.groq.com/openai/v1/chat/completions"
 TEXT_MODEL = "llama-3.1-8b-instant"
 
 MODE_PROMPTS = {
-    "Normal Chat": (
-        "You are NeoHelper, Eyad’s branded assistant. "
-        "Be concise, friendly, and helpful."
-    ),
-    "School Helper": (
-        "You are NeoHelper, a school helper for a Grade 5 student. "
-        "Explain clearly, step by step, using simple language."
-    ),
-    "Shopping Assistant": (
-        "You are NeoHelper, a shopping assistant in Saudi Arabia. "
-        "Compare specs, prices, and value clearly and briefly."
-    ),
-    "Gaming Help": (
-        "You are NeoHelper, an energetic gaming helper. "
-        "Give practical tips, fixes, and short, clear steps."
-    ),
+    "Normal": "You are NeoHelper, a helpful, concise assistant.",
+    "School": "You are NeoHelper, explaining clearly for a Grade 5 student.",
+    "Shopping": "You are NeoHelper, a shopping assistant in Saudi Arabia.",
+    "Gaming": "You are NeoHelper, a practical gaming helper.",
 }
 
-def extract_images_from_pdf(pdf_path: str):
-    try:
-        return convert_from_path(pdf_path)
-    except:
-        return None
-def call_groq_text(message, system_prompt):
-    try:
-        resp = groq_client.chat.completions.create(
-            model=TEXT_MODEL,
-            messages=[
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": message},
-            ],
-            max_tokens=900,
-        )
-        return resp.choices[0].message.content
-    except Exception as e:
-        return f"⚠️ Text model error: {str(e)}"
-
-def summarize_pdf_analyses(analyses, system_prompt):
-    joined = "\n\n--- PAGE BREAK ---\n\n".join(analyses)
-    prompt = (
-        "Summarize these PDF page analyses into one clear explanation for a student. "
-        "Be structured, simple, and concise:\n\n"
-        f"{joined}"
-    )
-    return call_groq_text(prompt, system_prompt)
-    
-def resolve_gradio_file(file):
-    if file is None:
-        return None, None
-
-    if isinstance(file, dict):
-        path = file.get("path") or file.get("name")
-        name = file.get("orig_name") or path
-        return path, name
-
-    path = getattr(file, "name", None)
-    name = getattr(file, "orig_name", path)
-    return path, name
-
-def neohelper_chat(message, history, file, mode):
-    if history is None:
-        history = []
-
-    system_prompt = MODE_PROMPTS.get(mode, MODE_PROMPTS["Normal Chat"])
-
-    if message is None:
-        message = ""
-    message = message.strip()
-
-    if isinstance(file, list):
-        file = file[0] if file else None
-
-    file_path, orig_name = resolve_gradio_file(file)
-
-    # TEXT ONLY
-    if not file_path:
-        reply = call_groq_text(message, system_prompt) if message else "Please type a question or upload a file."
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": reply})
-        return history, ""
-
-    ext = (orig_name or file_path).lower()
-
-    # IMAGE
-    if ext.endswith((".png", ".jpg", ".jpeg", ".webp", ".bmp")):
-        try:
-            prompt = message or "Explain this image in a simple way for a student."
-            img = Image.open(file_path).convert("RGB")
-            reply = analyze_image(img, prompt)
-        except Exception as e:
-            reply = f"⚠️ Error analyzing image: {str(e)}"
-
-        history.append({"role": "user", "content": message or f"[Image: {orig_name}]"})
-        history.append({"role": "assistant", "content": reply})
-        return history, ""
-
-    # PDF
-    if ext.endswith(".pdf"):
-        try:
-            pages = extract_images_from_pdf(file_path)
-            if not pages:
-                reply = "This PDF contains no pages I can analyze."
-            else:
-                analyses = []
-                for idx, img in enumerate(pages, start=1):
-                    prompt = message or f"Explain page {idx} simply."
-                    analyses.append(f"Page {idx}:\n{analyze_image(img, prompt)}")
-                reply = summarize_pdf_analyses(analyses, system_prompt)
-        except Exception as e:
-            reply = f"⚠️ Error processing PDF: {str(e)}"
-
-        history.append({"role": "user", "content": message or f"[PDF: {orig_name}]"})
-        history.append({"role": "assistant", "content": reply})
-        return history, ""
-
-    reply = "Unsupported file type. Please upload an image or a PDF."
-    history.append({"role": "user", "content": message})
-    history.append({"role": "assistant", "content": reply})
-    return history, ""
-
+# -----------------------------
+# GROQ CHAT
+# -----------------------------
+def groq_chat(user_msg, history, mode):
+    system_prompt = MODE_PROMPTS.get(mode, MODE_PROMPTS["Normal"])
+
+    messages = [{"role": "system", "content": system_prompt}]
+    if history:
+        for user, bot in history:
+            if user:
+                messages.append({"role": "user", "content": user})
+            if bot:
+                messages.append({"role": "assistant", "content": bot})
+    messages.append({"role": "user", "content": user_msg})
+
+    payload = {
+        "model": TEXT_MODEL,
+        "messages": messages,
+        "max_tokens": 400,
+        "temperature": 0.4,
+    }
+
+    headers = {
+        "Authorization": f"Bearer {GROQ_API_KEY}",
+        "Content-Type": "application/json",
+    }
+
+    resp = requests.post(GROQ_URL, json=payload, headers=headers, timeout=30)
+    resp.raise_for_status()
+    data = resp.json()
+    return data["choices"][0]["message"]["content"]
+
+# -----------------------------
+# GRADIO UI
+# -----------------------------
 def build_ui():
-    with gr.Blocks(title="NeoHelper") as demo:
-        gr.Markdown("# 🧠 NeoHelper")
+    with gr.Blocks(title="NeoHelper (Text Only)") as demo:
+        gr.Markdown("# **NeoHelper — Text‑Only (Groq)**")
 
         with gr.Row():
+            with gr.Column(scale=3):
+                chatbot = gr.Chatbot(height=520, label="NeoHelper")
+                with gr.Row():
+                    user_input = gr.Textbox(
+                        placeholder="Type your message…",
+                        lines=3,
+                        show_label=False,
+                    )
+                    send_btn = gr.Button("Send", scale=1)
+                clear_btn = gr.Button("Clear chat")
+
             with gr.Column(scale=1):
-                mode_dd = gr.Dropdown(
-                    choices=list(MODE_PROMPTS.keys()),
-                    value="Normal Chat",
+                mode = gr.Radio(
+                    ["Normal", "School", "Shopping", "Gaming"],
+                    value="Normal",
                     label="Mode",
                 )
-                file_input = gr.File(
-                    label="Upload image or PDF",
-                    file_types=["image", ".pdf"],
-                    file_count="single",
-                )
 
-            with gr.Column(scale=3):
-                chatbot = gr.Chatbot(
-                    label="NeoHelper",
-                    show_label=False,
-                    height=500,
-                )
-                msg = gr.Textbox(label="Message", lines=3)
-                send_btn = gr.Button("Send")
-                clear_btn = gr.Button("Clear")
+        def on_send(msg, history, mode):
+            history = history or []
+            if not msg.strip():
+                return history, ""
+            reply = groq_chat(msg.strip(), history, mode)
+            history.append([msg, reply])
+            return history, ""
 
         send_btn.click(
-            neohelper_chat,
-            inputs=[msg, chatbot, file_input, mode_dd],
-            outputs=[chatbot, msg],
+            on_send,
+            inputs=[user_input, chatbot, mode],
+            outputs=[chatbot, user_input],
         )
 
-        clear_btn.click(lambda: ([], ""), None, [chatbot, msg])
+        clear_btn.click(lambda: ([], ""), None, [chatbot, user_input])
 
     return demo
 
+app = build_ui()
+
 if __name__ == "__main__":
-    app = build_ui()
-    app.launch(share=False, ssr_mode=False)
+    app.launch(server_name="0.0.0.0", server_port=7860, theme="gradio/soft")