Update app.py

app.py

@@ -5,6 +5,7 @@ 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
 
 # Initialize session manager and get HF API key
 session_manager = SessionManager()
@@ -130,7 +131,8 @@ with gr.Blocks() as demo:
         new_session = gr.Button("🔄 New Session")
 
     chatbot = gr.Chatbot(height=600)
-    msg = gr.Textbox(label="Message")
+    msg = TextboxWithUpload(label="Message")
+    msg.add_upload_button()
     save_history = gr.Checkbox(label="Save Conversation History", value=True)
 
     def on_new_session():
@@ -146,6 +148,11 @@ 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):
@@ -155,9 +162,14 @@ with gr.Blocks() as demo:
                 history[-1][1] = response
                 yield history
 
+    def process_upload(file):
+        return f"Uploaded file: {file.name}"
+
+    # 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(
         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__":