Update app.py

app.py

@@ -80,7 +80,7 @@ def predict_eye_disease(input_image):
 def doctor_space(patient_id):
     for patient in patients_db:
         if patient["ID"] == patient_id:
-            return f"⚠️ Precautions: {patient['Precautions']}\n👩‍⚕️ Recommended Doctor: {patient['Doctor']}"
+            return f"⚠ Precautions: {patient['Precautions']}\n👩‍⚕ Recommended Doctor: {patient['Doctor']}"
     return "❌ Patient not found. Please check the ID."
 
 def pharmacist_space(patient_id):
@@ -95,8 +95,8 @@ def patient_dashboard(patient_id, password):
             return (f"🩺 Name: {patient['Name']}\n"
                     f"📋 Diagnosis: {patient['Diagnosis']}\n"
                     f"💊 Medications: {patient['Medications']}\n"
-                    f"⚠️ Precautions: {patient['Precautions']}\n"
-                    f"👩‍⚕️ Recommended Doctor: {patient['Doctor']}")
+                    f"⚠ Precautions: {patient['Precautions']}\n"
+                    f"👩‍⚕ Recommended Doctor: {patient['Doctor']}")
     return "❌ Access Denied: Invalid ID or Password."
 
 def doctor_dashboard(password):
@@ -109,25 +109,68 @@ def doctor_dashboard(password):
         details.append(f"🩺 Name: {patient['Name']}\n"
                        f"📋 Diagnosis: {patient['Diagnosis']}\n"
                        f"💊 Medications: {patient['Medications']}\n"
-                       f"⚠️ Precautions: {patient['Precautions']}\n"
-                       f"👩‍⚕️ Recommended Doctor: {patient['Doctor']}")
+                       f"⚠ Precautions: {patient['Precautions']}\n"
+                       f"👩‍⚕ Recommended Doctor: {patient['Doctor']}")
     return "\n\n".join(details)
 
-# Load the doctor consultant model
-doctor_consultant = pipeline("text-generation", model="ahmed-7124/dgptAW")
+# Gradio Interfaces
+registration_interface = gr.Interface(
+    fn=register_patient,
+    inputs=[
+        gr.Textbox(label="Patient Name"),
+        gr.Number(label="Age"),
+        gr.Radio(label="Gender", choices=["Male", "Female", "Other"]),
+        gr.Textbox(label="Set Password", type="password"),
+    ],
+    outputs="text",
+)
+
+pdf_extraction_interface = gr.Interface(
+    fn=extract_pdf_report,
+    inputs=gr.File(label="Upload PDF Report"),
+    outputs="text",
+)
+
+report_analysis_interface = gr.Interface(
+    fn=analyze_report,
+    inputs=[
+        gr.Number(label="Patient ID"),
+        gr.Textbox(label="Report Text"),
+    ],
+    outputs="text",
+)
+
+eye_disease_interface = gr.Interface(
+    fn=predict_eye_disease,
+    inputs=gr.Image(label="Upload an Eye Image", type="numpy"),
+    outputs="text",
+)
+
+doctor_space_interface = gr.Interface(
+    fn=doctor_space,
+    inputs=gr.Number(label="Patient ID"),
+    outputs="text",
+)
+
+pharmacist_space_interface = gr.Interface(
+    fn=pharmacist_space,
+    inputs=gr.Number(label="Patient ID"),
+    outputs="text",
+)
 
-# New function for Doctor Consultant (Text Generation for Consultation)
-def doctor_consultation(query):
-    response = doctor_consultant(query, max_length=200, num_return_sequences=1)
-    return response[0]['generated_text']
+patient_dashboard_interface = gr.Interface(
+    fn=patient_dashboard,
+    inputs=[
+        gr.Number(label="Patient ID"),
+        gr.Textbox(label="Password", type="password"),
+    ],
+    outputs="text",
+)
 
-# Adding the doctor consultant interface to Gradio
-doctor_consultation_interface = gr.Interface(
-    fn=doctor_consultation,
-    inputs=gr.Textbox(label="Ask the Doctor (Consultation Query)", placeholder="Enter your medical query here..."),
+doctor_dashboard_interface = gr.Interface(
+    fn=doctor_dashboard,
+    inputs=gr.Textbox(label="Doctor Password", type="password"),
     outputs="text",
-    title="Doctor Consultant",
-    description="Get expert advice from the Doctor Consultant powered by AI. Ask any medical-related questions and get responses."
 )
 
 # Gradio App Layout
@@ -157,8 +200,5 @@ with gr.Blocks() as app:
     
     with gr.Tab("Doctor Dashboard"):
         doctor_dashboard_interface.render()
-    
-    with gr.Tab("Doctor Consultant"):
-        doctor_consultation_interface.render()
 
-app.launch(share=True)
+app.launch(share=True)