Update app.py

app.py

@@ -74,6 +74,37 @@ def get_shap_data(input_text):
   html_shap_content = shap.plots.text(shap_values, display=False)
   return html_shap_content
 
+all_options = {
+  'Heart hurts': "Immediate medical attention is required. Please seek emergency care.",
+  'Acne': "Consider dermatological treatment options such as topical creams, medications, or lifestyle changes.",
+  'Hair falling out': "Consult a dermatologist to address hair-related concerns.",
+  'Infected wound': "Keep the wound clean and dry. If signs of infection develop, seek immediate medical attention. Consult with a surgeon for evaluation.",
+  'Skin issue': "Seek advice from a dermatologist for personalized treatment options.",
+  'Cough': "Stay hydrated, get plenty of rest, and consider over-the-counter cough medications.",
+  'Ear ache': "Avoid inserting anything into the ear and consult an ENT specialist for proper examination and treatment.",
+  'Feeling cold': "Rest, stay warm, and drink plenty of fluids. Seek medical attention if symptoms worsen.",
+  'Stomach ache': "Avoid spicy or fatty foods, and consider over-the-counter antacids for temporary relief.",
+  'Internal pain': "It's important to get a proper diagnosis. Please consult with a gastroenterologist for further evaluation.",
+  'Open wound': "Keep the wound clean and dry. If signs of infection develop, seek immediate medical attention. Consult with a surgeon for evaluation.",
+  'Feeling dizzy': "Ensure you are well-hydrated and rested. If symptoms persist, consult with a general physician for evaluation.",
+  'Body feels weak': "Get plenty of rest, eat a balanced diet, and consider consulting with a general physician for further evaluation.",
+  'Head ache': "Ensure you are well-hydrated and rested. If headaches persist, consult with a neurologist for evaluation.", 
+  'Blurry vision': "Ensure you are well-rested and take regular breaks from screens. If vision problems persist, consult with an ophthalmologist for evaluation.",
+  'Joint pain': "Rest, gentle stretching, and over-the-counter pain relievers may provide relief. If symptoms persist, consult with an orthopedist for evaluation.",
+  'Knee pain': "Rest, gentle stretching, and over-the-counter pain relievers may provide relief. If symptoms persist, consult with an orthopedist for evaluation.",
+  'Back pain': "Rest, gentle stretching, and over-the-counter pain relievers may provide relief. If symptoms persist, consult with an orthopedist for evaluation.",
+  'Emotional pain': "Seek support from friends and family, and consider speaking with a therapist or psychiatrist for evaluation and support.",
+  'Hard to breath': "If experiencing difficulty breathing, seek immediate medical attention. Consult with a respirologist for evaluation.",
+  'Muscle pain': "Rest, gentle exercise, and over-the-counter pain relievers may provide relief. If symptoms persist, consult with a rheumatologist for evaluation.",
+  'Injury from sports': "Rest, gentle exercise, and over-the-counter pain relievers may provide relief. If symptoms persist, consult with a rheumatologist for evaluation.",
+  'Foot ache': "Rest, gentle exercise, and over-the-counter pain relievers may provide relief. If symptoms persist, consult with a rheumatologist for evaluation.",
+  'Shoulder pain': "Rest, gentle stretching, and over-the-counter pain relievers may provide relief. If symptoms persist, consult with a rheumatologist or gastroenterologist for evaluation.",
+  'Neck pain': "Rest, gentle stretching, and over-the-counter pain relievers may provide relief. If symptoms persist, consult with a rheumatologist or orthopedist for evaluation.",
+}
+
+def pt_f(x):
+  return all_options.get(x)
+
 css = """
 textarea {
     background-color: #00000000;
@@ -108,7 +139,29 @@ with gr.Blocks(title="SpecX", css=css, theme=gr.themes.Soft()) as demo:
     submit_btn_shap.click(get_shap_data, inputs=text_box, outputs=shap_content_box)
   with gr.Row():
     textmd1 = gr.Markdown('''
+
       <div style="margin: 50px 0;"></div>
+
+    ''')
+  with gr.Column():
+    textmd1 = gr.Markdown('''
+    ## Personalised Treatment
+    ''')
+    with gr.Row():
+      with gr.Column():
+        specialist_symptom = gr.Dropdown(
+            label="Select Specialist Type",
+            choices=list(all_options.keys()),
+        )
+        submit_btn2 = gr.Button("Submit", elem_id="warningk", variant='primary')
+      with gr.Column():
+        pt_text = gr.Textbox(label="Recommendation")
+      submit_btn2.click(pt_f, inputs=specialist_symptom, outputs=pt_text)
+  with gr.Row():
+    textmd1 = gr.Markdown('''
+
+      <div style="margin: 50px 0;"></div>
+
     ''')
   with gr.Row():
     with gr.Column():