Update app.py

app.py

@@ -2,20 +2,27 @@ import gradio as gr
 import requests
 import mimetypes
 import os
+import threading
+import uvicorn
 import re
 
-# === Fix surrogate character issue ===
 def sanitize(text):
     if isinstance(text, str):
         text = re.sub(r'[\uD800-\uDFFF]', '', text)
         return text.encode("utf-8", "ignore").decode("utf-8")
     return text
 
-# API endpoints (your FastAPI runs in main.py separately)
-FILE_API_URL = "/chat-with-file"
-URL_API_URL = "/chat-with-url"
-IMAGE_API_URL = "/extract-text-from-image"
-AUDIO_API_URL = "/transcribe-audio"
+def run_fastapi():
+    uvicorn.run("main:app", host="0.0.0.0", port=8000)
+
+threading.Thread(target=run_fastapi, daemon=True).start()
+
+API_URLS = {
+    "file": "http://127.0.0.1:8000/chat-with-file",
+    "url": "http://127.0.0.1:8000/chat-with-url",
+    "image": "http://127.0.0.1:8000/extract-text-from-image",
+    "audio": "http://127.0.0.1:8000/transcribe-audio"
+}
 
 chat_sessions = {"file": [], "url": []}
 
@@ -24,137 +31,86 @@ def format_chat(history):
 
 def ask_from_file(file_obj, question, session_id):
     if not file_obj or not question.strip():
-        return "⚠️ Please upload a file and enter a question.", chat_sessions["file"]
+        return "⚠️ Upload a file and enter a question.", chat_sessions["file"]
     if os.path.getsize(file_obj.name) > 10 * 1024 * 1024:
-        return "❌ File too large. Please upload files under 10MB.", chat_sessions["file"]
-
+        return "❌ File too large. Max 10MB.", chat_sessions["file"]
     try:
-        mime_type, _ = mimetypes.guess_type(file_obj.name)
+        mime, _ = mimetypes.guess_type(file_obj.name)
         with open(file_obj.name, "rb") as f:
-            files = {"file": (file_obj.name, f, mime_type or "application/octet-stream")}
+            files = {"file": (file_obj.name, f, mime or "application/octet-stream")}
             history = chat_sessions["file"]
             context = "\n".join([f"Q: {q}\nA: {a}" for q, a in history]) + f"\nQ: {question}"
-            data = {"question": context}
-            response = requests.post(FILE_API_URL, files=files, data=data)
-            answer = sanitize(response.json().get("answer", "⚠️ No answer returned."))
+            res = requests.post(API_URLS["file"], files=files, data={"question": context})
+            answer = sanitize(res.json().get("answer", "⚠️ No answer."))
             chat_sessions["file"].append((question, answer))
             return format_chat(chat_sessions["file"]), chat_sessions["file"]
     except Exception as e:
-        return f"❌ Error: {str(e)}", chat_sessions["file"]
+        return f"❌ Error: {e}", chat_sessions["file"]
 
 def ask_from_url(url, question, session_id):
     if not url.strip() or not question.strip():
-        return "⚠️ Please enter both a valid URL and a question.", chat_sessions["url"]
+        return "⚠️ URL and question required.", chat_sessions["url"]
     try:
         history = chat_sessions["url"]
         context = "\n".join([f"Q: {q}\nA: {a}" for q, a in history]) + f"\nQ: {question}"
-        response = requests.post(URL_API_URL, json={"url": url, "question": context})
-        answer = sanitize(response.json().get("answer", "⚠️ No answer returned."))
+        res = requests.post(API_URLS["url"], json={"url": url, "question": context})
+        answer = sanitize(res.json().get("answer", "⚠️ No answer."))
         chat_sessions["url"].append((question, answer))
         return format_chat(chat_sessions["url"]), chat_sessions["url"]
     except Exception as e:
-        return f"❌ Error: {str(e)}", chat_sessions["url"]
+        return f"❌ Error: {e}", chat_sessions["url"]
 
 def extract_text_from_image(image_path):
     try:
         with open(image_path, "rb") as f:
             files = {"file": ("image.png", f, "image/png")}
-            response = requests.post(IMAGE_API_URL, files=files)
-            return sanitize(response.json().get("answer", "⚠️ No text extracted."))
+            res = requests.post(API_URLS["image"], files=files)
+            return sanitize(res.json().get("answer", "⚠️ No text extracted."))
     except Exception as e:
-        return f"❌ Error: {str(e)}"
+        return f"❌ Error: {e}"
 
 def transcribe_audio(audio_path):
     try:
         with open(audio_path, "rb") as f:
             files = {"file": ("audio.wav", f, "audio/wav")}
-            response = requests.post(AUDIO_API_URL, files=files)
-            return sanitize(response.json().get("answer", "⚠️ No transcript returned."))
+            res = requests.post(API_URLS["audio"], files=files)
+            return sanitize(res.json().get("answer", "⚠️ No transcript returned."))
     except Exception as e:
-        return f"❌ Error: {str(e)}"
-
-def clear_file_chat():
-    chat_sessions["file"] = []
-    return "", chat_sessions["file"]
-
-def clear_url_chat():
-    chat_sessions["url"] = []
-    return "", chat_sessions["url"]
-
-# 💅 CSS
-custom_css = """
-body, .gradio-container {
-    background-color: #111111 !important;
-    color: white !important;
-}
-footer { display: none !important; }
-h1 {
-    font-size: 2.2em;
-    font-weight: bold;
-    text-align: center;
-    color: white;
-    margin-bottom: 20px;
-}
-.gr-button {
-    background-color: #00FF88;
-    color: black;
-    border: none;
-    border-radius: 6px;
-    font-weight: bold;
-}
-.gr-button:hover {
-    background-color: #00cc70;
-}
-.gr-input, .gr-textbox, .gr-file, textarea, input {
-    background-color: #1e1e1e !important;
-    color: white !important;
-    border: 1px solid #00FF88 !important;
-    border-radius: 6px !important;
-}
-.gr-tabitem[data-selected="true"] > button {
-    background-color: #00FF88 !important;
-    color: black !important;
-}
-.gr-tabitem > button {
-    background-color: transparent !important;
-    color: white !important;
-}
-"""
-
-# UI
-with gr.Blocks(css=custom_css) as demo:
-    gr.Markdown("# 🤖 AI Chatbot with File, Web, Image, Audio & Chat History")
-
-    with gr.Tab("📂 Chat with File"):
-        file_input = gr.File(label="Upload File (.txt, .csv, .docx, .pdf)", file_types=[".txt", ".csv", ".docx", ".pdf"])
-        file_question = gr.Textbox(label="Your Question", placeholder="Ask something based on file")
-        file_button = gr.Button("Ask from File")
-        file_output = gr.Textbox(label="Chat History", lines=10)
-        clear_file = gr.Button("New File Chat")
-
-        file_button.click(fn=ask_from_file, inputs=[file_input, file_question, gr.State("file")], outputs=[file_output, gr.State("file")])
-        clear_file.click(fn=clear_file_chat, inputs=[], outputs=[file_output, gr.State("file")])
-
-    with gr.Tab("🌐 Chat with Website"):
-        url_input = gr.Textbox(label="Website URL", placeholder="https://example.com")
-        url_question = gr.Textbox(label="Your Question", placeholder="Ask something based on webpage")
-        url_button = gr.Button("Ask from URL")
-        url_output = gr.Textbox(label="Chat History", lines=10)
-        clear_url = gr.Button("New URL Chat")
-
-        url_button.click(fn=ask_from_url, inputs=[url_input, url_question, gr.State("url")], outputs=[url_output, gr.State("url")])
-        clear_url.click(fn=clear_url_chat, inputs=[], outputs=[url_output, gr.State("url")])
-
-    with gr.Tab("🖼️ Extract Text from Image"):
-        image_input = gr.Image(label="Upload Image", type="filepath")
-        image_button = gr.Button("Extract Text")
-        image_output = gr.Textbox(label="Extracted Text", lines=8)
-        image_button.click(fn=extract_text_from_image, inputs=image_input, outputs=image_output)
-
-    with gr.Tab("🎤 Transcribe Audio"):
-        audio_input = gr.Audio(label="Upload Audio", type="filepath")
-        audio_button = gr.Button("Transcribe Audio")
-        audio_output = gr.Textbox(label="Transcript", lines=8)
-        audio_button.click(fn=transcribe_audio, inputs=audio_input, outputs=audio_output)
+        return f"❌ Error: {e}"
+
+def clear_file_chat(): chat_sessions["file"] = []; return "", chat_sessions["file"]
+def clear_url_chat(): chat_sessions["url"] = []; return "", chat_sessions["url"]
+
+with gr.Blocks(css="") as demo:
+    gr.Markdown("# 🤖 AI Chatbot with File, Web, Image, Audio")
+    with gr.Tab("📂 File"):
+        file = gr.File(file_types=[".txt", ".csv", ".docx", ".pdf"])
+        q1 = gr.Textbox(label="Question")
+        b1 = gr.Button("Ask from File")
+        out1 = gr.Textbox(label="Chat", lines=10)
+        clr1 = gr.Button("Clear")
+        b1.click(fn=ask_from_file, inputs=[file, q1, gr.State("file")], outputs=[out1, gr.State("file")])
+        clr1.click(fn=clear_file_chat, outputs=[out1, gr.State("file")])
+
+    with gr.Tab("🌐 URL"):
+        url = gr.Textbox(label="Website URL")
+        q2 = gr.Textbox(label="Question")
+        b2 = gr.Button("Ask from URL")
+        out2 = gr.Textbox(label="Chat", lines=10)
+        clr2 = gr.Button("Clear")
+        b2.click(fn=ask_from_url, inputs=[url, q2, gr.State("url")], outputs=[out2, gr.State("url")])
+        clr2.click(fn=clear_url_chat, outputs=[out2, gr.State("url")])
+
+    with gr.Tab("🖼️ Image OCR"):
+        img = gr.Image(type="filepath")
+        b3 = gr.Button("Extract Text")
+        out3 = gr.Textbox(label="Text", lines=8)
+        b3.click(fn=extract_text_from_image, inputs=img, outputs=out3)
+
+    with gr.Tab("🎤 Audio"):
+        aud = gr.Audio(type="filepath")
+        b4 = gr.Button("Transcribe")
+        out4 = gr.Textbox(label="Transcript", lines=8)
+        b4.click(fn=transcribe_audio, inputs=aud, outputs=out4)
 
 demo.launch(share=False)