Update app.py

app.py

@@ -1,39 +1,3 @@
-import os, uuid, requests, gradio as gr, pdfplumber, docx, pandas as pd
-from PIL import Image
-import whisper
-
-GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
-MODEL_NAME = "llama3-8b-8192"  # Update if using a different model
-
-def extract_text_from_file(file):
-    if file is None: return ""
-    ext = os.path.splitext(file.name)[-1].lower()
-    try:
-        if ext == ".pdf":
-            with pdfplumber.open(file.name) as pdf:
-                return "\n".join([page.extract_text() or "" for page in pdf.pages])
-        elif ext == ".docx":
-            return "\n".join([p.text for p in docx.Document(file.name).paragraphs])
-        elif ext in [".xls", ".xlsx", ".csv"]:
-            df = pd.read_excel(file.name) if ext != ".csv" else pd.read_csv(file.name)
-            return df.to_string(index=False)
-        elif ext.endswith((".png", ".jpg", ".jpeg", ".bmp")):
-            return "Image uploaded. You can ask about it."
-        elif ext.endswith((".txt", ".py", ".json", ".html", ".md", ".css", ".js")):
-            with open(file.name, encoding="utf-8", errors="ignore") as f: return f.read()
-    except Exception as e:
-        return f"File extraction failed: {e}"
-
-def transcribe_audio(audio_path):
-    try:
-        if not audio_path: return ""
-        model = whisper.load_model("base")
-        result = model.transcribe(audio_path)
-        return result["text"]
-    except Exception as e:
-        return f"Audio transcription failed: {e}"
-
 def chat_response(history, message):
     try:
         headers = {
@@ -51,92 +15,10 @@ def chat_response(history, message):
         }
 
         response = requests.post(GROQ_API_URL, headers=headers, json=payload)
-        reply = response.json()["choices"][0]["message"]["content"]
-        return reply.strip()
-    except Exception as e:
-        return f"Groq response failed: {e}"
+        response.raise_for_status()  # catches HTTP errors
+        return response.json()["choices"][0]["message"]["content"].strip()
 
-def process_input(msg, file, audio, history):
-    file_text = extract_text_from_file(file) if file else ""
-    audio_text = transcribe_audio(audio) if audio else ""
-    full_input = "\n".join([i for i in [msg, file_text, audio_text] if i])
-    if not full_input.strip():
-        return history, gr.Textbox.update(value=""), None, None, gr.Textbox.update(value="No input provided.")
-    reply = chat_response(history, full_input)
-    history.append(("user", msg.strip() or file_text or audio_text))
-    history.append(("assistant", reply))
-    return history, gr.Textbox.update(value=""), None, None, gr.Textbox.update(value="Response generated.")
-
-def save_chat_history(history):
-    try:
-        os.makedirs("saved_chats", exist_ok=True)
-        title = history[-2][1][:30].replace(" ", "_") if len(history) >= 2 else str(uuid.uuid4())
-        filename = f"saved_chats/{title}.txt"
-        with open(filename, "w", encoding="utf-8") as f:
-            for role, text in history:
-                f.write(f"{role}: {text}\n")
-        return gr.Textbox.update(value="Chat saved successfully.")
+    except requests.exceptions.HTTPError as http_err:
+        return f"HTTP error: {http_err}\nResponse: {response.text}"
     except Exception as e:
-        return gr.Textbox.update(value=f"Failed to save chat: {e}")
-
-def get_saved_chats():
-    if not os.path.exists("saved_chats"): return []
-    return sorted([f for f in os.listdir("saved_chats") if f.endswith(".txt")])
-
-def load_selected_chat(filename):
-    filepath = os.path.join("saved_chats", filename)
-    if not os.path.exists(filepath): return [], gr.Textbox.update(value="File not found.")
-    history = []
-    with open(filepath, "r", encoding="utf-8") as f:
-        for line in f:
-            if ":" in line:
-                role, content = line.strip().split(":", 1)
-                history.append((role.strip(), content.strip()))
-    return history, gr.Textbox.update(value=f"Chat '{filename}' loaded.")
-
-def clear_chat(): return [], gr.Textbox.update(value="Chat cleared. Start fresh!")
-
-custom_css = """
-.gradio-container { max-width: 1200px !important; margin: auto !important; }
-select { appearance: auto !important; }
-"""
-
-with gr.Blocks(css=custom_css, title="Neobot - Chatbot") as demo:
-    gr.Markdown("# 🤖 Neobot - Advanced AI Chatbot")
-
-    chatbot = gr.Chatbot(label="Chat History", height=500, show_copy_button=True)
-    message_input = gr.Textbox(placeholder="Type your message or upload a file...", scale=4)
-    send_button = gr.Button("Send 📤", scale=1)
-
-    file_upload = gr.File(label="📎 Upload File", file_types=[".pdf", ".docx", ".txt", ".xlsx", ".xls", ".csv", ".png", ".jpg", ".jpeg", ".gif", ".bmp", ".py", ".js", ".html", ".css", ".json", ".md"])
-    audio_upload = gr.Audio(label="🎤 Upload Audio", type="filepath")
-
-    save_button = gr.Button("💾 Save Chat")
-    clear_button = gr.Button("🆕 New Chat")
-    refresh_button = gr.Button("🔄 Refresh List")
-
-    saved_chats_dropdown = gr.Dropdown(label="📂 Load Saved Chat", choices=get_saved_chats(), type="value")
-    load_button = gr.Button("📂 Load Chat")
-
-    status_message = gr.Textbox(label="Status", interactive=False)
-    chat_history = gr.State([])
-
-    send_button.click(process_input, [message_input, file_upload, audio_upload, chat_history], [chat_history, message_input, file_upload, audio_upload, status_message])\
-               .then(lambda h: h, [chat_history], [chatbot])
-
-    message_input.submit(process_input, [message_input, file_upload, audio_upload, chat_history], [chat_history, message_input, file_upload, audio_upload, status_message])\
-                 .then(lambda h: h, [chat_history], [chatbot])
-
-    save_button.click(save_chat_history, [chat_history], [status_message])\
-               .then(lambda: gr.Dropdown.update(choices=get_saved_chats()), None, [saved_chats_dropdown])
-
-    clear_button.click(clear_chat, None, [chat_history, status_message])\
-                .then(lambda: [], None, [chatbot])
-
-    refresh_button.click(lambda: gr.Dropdown.update(choices=get_saved_chats()), None, [saved_chats_dropdown])
-
-    load_button.click(load_selected_chat, [saved_chats_dropdown], [chat_history, status_message])\
-               .then(lambda h: h, [chat_history], [chatbot])
-
-if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=False, debug=True)
+        return f"Groq response failed: {e}"