Update app.py

app.py

@@ -9,12 +9,12 @@ sample_questions = [
     "What are the first-line medications for treating hypertension?",
 ]
 
-def qa_bot(user_question, history):
+def qa_bot(user_question, history, year, specialty):
     history = history or []
     if not user_question.strip():
         return history, history
     history.append({"role": "user", "content": user_question})
-    result = rag_chain.forward(user_question)
+    result = rag_chain.forward(user_question, year, specialty)
     answer = result.final_answer
     history.append({"role": "assistant", "content": answer})
     return history, history
@@ -107,6 +107,24 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
                 btn = gr.Button(q, elem_id=f"suggestion-{i}", elem_classes=["suggestion-btn"])
                 suggestion_buttons.append(btn)
 
+        with gr.Row():
+            specialty_options = [
+                "Rheumatology", "Psychiatry", "Pulmonology & Respiratory Medicine", "Nephrology", "Public Health & Epidemiology",
+                "Medical Research & Methodology", "Pharmacy & Pharmacology", "Hematology", "Oncology", "Medical Ethics & Law",
+                "Medical Technology & Informatics", "Infectious Disease", "Basic Medical Sciences", "Allergology", "Geriatrics",
+                "Cardiology", "Gastroenterology & Hepatology", "General Surgery", "General Pediatrics", "Endocrinology & Metabolism",
+                "Vascular Surgery", "Radiology & Imaging", "Obstetrics & Gynecology", "Orthopedic Surgery", "Neurology",
+                "Family Medicine & Primary Care", "Psychology & Behavioral Health", "Otorhinolaryngology (ENT)", "General Internal Medicine",
+                "Anesthesiology", "Physical & Rehabilitation Medicine", "Medical Education", "Healthcare Administration & Management",
+                "Non-Medical Sciences & Disciplines", "Dermatology", "Critical Care & Intensive Care", "Urology", "Complementary & Alternative Medicine",
+                "Cardiothoracic Surgery", "Neurosurgery", "Pediatric Subspecialties", "Occupational & Environmental Health", "Ophthalmology",
+                "Emergency Medicine", "Dental & Oral Medicine", "Biomedical Engineering", "Pathology & Laboratory Medicine", "Transplant Surgery",
+                "Preventive Medicine", "Genetics", "Nursing", "Allied Health Professions", "Plastic & Reconstructive Surgery", "Others",
+                "Toxicology", "General Medicine"
+            ]
+            year_options = [str(y) for y in range(2021, 1792, -1)]
+            specialty_dropdown = gr.Dropdown(choices=specialty_options, value="General Medicine", label="Specialty", scale=4)
+            year_dropdown = gr.Dropdown(choices=year_options, value="2021", label="Year", scale=2)
         with gr.Row():
             user_input = gr.Textbox(placeholder="Type a medical question...", show_label=False, lines=1, scale=8, elem_id="user-input")
             submit_btn = gr.Button(value="➤", scale=1, elem_id="submit-btn")
@@ -114,8 +132,8 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
     def suggestion_click(q, history):
         return qa_bot(q, history)
 
-    submit_btn.click(qa_bot, inputs=[user_input, state], outputs=[chatbot, state])
-    user_input.submit(qa_bot, inputs=[user_input, state], outputs=[chatbot, state])
+    submit_btn.click(qa_bot, inputs=[user_input, state, year_dropdown, specialty_dropdown], outputs=[chatbot, state])
+    user_input.submit(qa_bot, inputs=[user_input, state, year_dropdown, specialty_dropdown], outputs=[chatbot, state])
     for btn, q in zip(suggestion_buttons, sample_questions):
         btn.click(suggestion_click, inputs=[gr.State(q), state], outputs=[chatbot, state])