Update app.py

app.py

@@ -1,19 +1,19 @@
 import gradio as gr
 import os
+import shutil
 import threading
 from datetime import datetime
 from typing import List, Dict, Any, Generator
 from session_manager import SessionManager
 from huggingface_hub import InferenceClient
-from textbox_with_upload import TextboxWithUpload
-
-# Check Gradio version
-print(f"Gradio version: {gr.__version__}")
 
 # Initialize session manager and get HF API key
 session_manager = SessionManager()
 HF_API_KEY = os.getenv("HF_API_KEY")
 
+# Create uploads directory if it doesn't exist
+os.makedirs("uploads", exist_ok=True)
+
 # Model endpoints configuration
 MODEL_ENDPOINTS = {
     "Qwen2.5-72B-Instruct": "https://api-inference.huggingface.co/models/Qwen/Qwen2.5-72B-Instruct",
@@ -133,15 +133,17 @@ with gr.Blocks() as demo:
         session_id = gr.State(session_manager.create_session)
         new_session = gr.Button("🔄 New Session")
 
-    chatbot = gr.Chatbot(height=600, type='messages')
-    msg = TextboxWithUpload(label="Message")
+    chatbot = gr.Chatbot(height=600)
     save_history = gr.Checkbox(label="Save Conversation History", value=True)
 
     def on_new_session():
         new_id = session_manager.create_session()
         return new_id, []
 
-    def user(message, history, session_id, save_history):
+    def user(message, file, history, session_id, save_history):
+        if file:
+            # Move uploaded file to uploads directory
+            message += f"\nUploaded file: {file}"
         if save_history:
             session = session_manager.load_session(session_id)
             session["history"].append({
@@ -150,11 +152,6 @@ with gr.Blocks() as demo:
                 "content": message
             })
             session_manager.save_session(session_id, session)
-        
-        # Handle file upload
-        if message.startswith("Uploaded file:"):
-            message = f"I've uploaded a file: {message.split(':', 1)[1].strip()}"
-        
         return "", history + [[message, None]]
 
     def bot(history, session_id):
@@ -164,15 +161,13 @@ with gr.Blocks() as demo:
                 history[-1][1] = response
                 yield history
 
-    def process_upload(file):
-        return f"Uploaded file: {file.name}"
+    with gr.Row():
+        msg = gr.Textbox(label="Message")
+        file_upload = gr.UploadButton("Upload file")
 
-    # Set up event handlers for message submission, file upload, and new session creation
-    msg.submit(user, [msg, chatbot, session_id, save_history], [msg, chatbot]).then(
+    msg.submit(user, [msg, file_upload, chatbot, session_id, save_history], [msg, chatbot]).then(
         bot, [chatbot, session_id], [chatbot]
     )
-    msg.upload_button.upload(process_upload, msg.upload_button, msg)  # Handle file uploads
     new_session.click(on_new_session, None, [session_id, chatbot])
-
 if __name__ == "__main__":
     demo.launch(share=True)