Upload app.py

app.py

@@ -0,0 +1,76 @@
+
+import gradio as gr
+import os
+from medical_chatbot import ColabBioGPTChatbot
+
+# Instantiate the chatbot
+chatbot = ColabBioGPTChatbot(use_gpu=True, use_8bit=True)
+
+medical_file_uploaded = False
+
+def upload_and_initialize(file):
+    global medical_file_uploaded
+
+    if file is None:
+        return (
+            "❌ Please upload a medical .txt file.",
+            gr.Chatbot.update(visible=False),
+            gr.Textbox.update(visible=False),
+            gr.Button.update(visible=False)
+        )
+    
+    try:
+        chatbot.load_medical_data(file.name)
+        medical_file_uploaded = True
+        return (
+            "✅ Medical data loaded and chatbot initialized!",
+            gr.Chatbot.update(visible=True),
+            gr.Textbox.update(visible=True),
+            gr.Button.update(visible=True)
+        )
+    except Exception as e:
+        return (
+            f"❌ Failed to load file: {str(e)}",
+            gr.Chatbot.update(visible=False),
+            gr.Textbox.update(visible=False),
+            gr.Button.update(visible=False)
+        )
+
+def generate_response(user_input):
+    if not medical_file_uploaded:
+        return "⚠️ Please upload and initialize medical data first."
+    return chatbot.chat(user_input)
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 🩺 Pediatric Medical Assistant\nUpload a medical .txt file and start chatting.")
+
+    with gr.Row():
+        file_input = gr.File(label="📁 Upload Medical File", file_types=[".txt"])
+        upload_btn = gr.Button("📤 Upload and Initialize")
+
+    upload_output = gr.Textbox(label="System Status", interactive=False)
+
+    chatbot_ui = gr.Chatbot(label="🧠 Chat History")
+    user_input = gr.Textbox(placeholder="Ask a pediatric health question...", lines=2, show_label=False)
+    submit_btn = gr.Button("Send")
+
+    chatbot_ui.visible = False
+    user_input.visible = False
+    submit_btn.visible = False
+
+    upload_btn.click(
+        fn=upload_and_initialize,
+        inputs=[file_input],
+        outputs=[upload_output, chatbot_ui, user_input, submit_btn]
+    )
+
+    def on_submit(user_message, chat_history):
+        bot_response = generate_response(user_message)
+        chat_history.append((user_message, bot_response))
+        return "", chat_history
+
+    user_input.submit(fn=on_submit, inputs=[user_input, chatbot_ui], outputs=[user_input, chatbot_ui])
+    submit_btn.click(fn=on_submit, inputs=[user_input, chatbot_ui], outputs=[user_input, chatbot_ui])
+
+demo.launch(share=True)
+