Update app.py

app.py

@@ -64,8 +64,7 @@ def extract_pdf_report_and_redirect(pdf):
     with pdfplumber.open(pdf.name) as pdf_file:
         for page in pdf_file.pages:
             text += page.extract_text()
-    # Redirect to the "Analyze Medical Report" tab and populate the extracted text
-    return text, gr.update(selected=True)
+    return text, gr.Tabs.update(selected="Analyze Medical Report")
 
 def predict_eye_disease(input_image):
     input_image = tf.image.resize(input_image, [224, 224]) / 255.0
@@ -114,106 +113,56 @@ def doctor_dashboard(password):
     return "\n\n".join(details)
 
 # Gradio App Layout
-def app_layout():
-    with gr.Blocks() as app:
-        gr.Markdown("# Medico GPT")
-        
-        with gr.Tab("Patient Registration"):
-            with gr.Row():
-                name = gr.Textbox(label="Patient Name")
-                age = gr.Number(label="Age")
-                gender = gr.Radio(label="Gender", choices=["Male", "Female", "Other"])
-                password = gr.Textbox(label="Set Password", type="password")
-                register_button = gr.Button("Register")
-                registration_output = gr.Textbox(label="Output", interactive=False)
-            
-            register_button.click(
-                fn=register_patient,
-                inputs=[name, age, gender, password],
-                outputs=registration_output,
-            )
-        
-        with gr.Tab("Analyze Medical Report") as analyze_tab:
-            patient_id = gr.Number(label="Patient ID")
-            report_text = gr.Textbox(label="Report Text")
-            analyze_button = gr.Button("Analyze")
-            analysis_output = gr.Textbox(label="Analysis Output", interactive=False)
-            
-            analyze_button.click(
-                fn=analyze_report,
-                inputs=[patient_id, report_text],
-                outputs=analysis_output,
-            )
-        
-        with gr.Tab("Extract PDF Report"):
-            pdf_file = gr.File(label="Upload PDF Report")
-            extract_button = gr.Button("Extract and Analyze")
-            
-            extract_button.click(
-                fn=extract_pdf_report_and_redirect,
-                inputs=pdf_file,
-                outputs=[
-                    report_text,  # Automatically populate the report text in the Analyze tab
-                    analyze_tab,  # Redirect to the Analyze Medical Report tab
-                ],
-            )
-        
-        with gr.Tab("Opthamologist Space"):
-            eye_image = gr.Image(label="Upload an Eye Image", type="numpy")
-            eye_result = gr.Textbox(label="Eye Disease Prediction", interactive=False)
-            predict_button = gr.Button("Predict")
-            
-            predict_button.click(
-                fn=predict_eye_disease,
-                inputs=eye_image,
-                outputs=eye_result,
-            )
-        
-        with gr.Tab("Doctor Space"):
-            doc_patient_id = gr.Number(label="Patient ID")
-            doctor_output = gr.Textbox(label="Doctor Details", interactive=False)
-            doctor_button = gr.Button("Get Details")
-            
-            doctor_button.click(
-                fn=doctor_space,
-                inputs=doc_patient_id,
-                outputs=doctor_output,
-            )
+with gr.Blocks() as app:
+    gr.Markdown("# Medico GPT")
+    
+    with gr.Tab("Patient Registration"):
+        name = gr.Textbox(label="Patient Name")
+        age = gr.Number(label="Age")
+        gender = gr.Radio(label="Gender", choices=["Male", "Female", "Other"])
+        password = gr.Textbox(label="Set Password", type="password")
+        register_button = gr.Button("Register")
+        register_output = gr.Textbox(label="Output")
         
-        with gr.Tab("Pharmacist Space"):
-            pharm_patient_id = gr.Number(label="Patient ID")
-            pharmacist_output = gr.Textbox(label="Pharmacist Details", interactive=False)
-            pharmacist_button = gr.Button("Get Details")
-            
-            pharmacist_button.click(
-                fn=pharmacist_space,
-                inputs=pharm_patient_id,
-                outputs=pharmacist_output,
-            )
+        register_button.click(
+            fn=register_patient,
+            inputs=[name, age, gender, password],
+            outputs=register_output
+        )
+    
+    with gr.Tab("Analyze Medical Report") as analyze_tab:
+        patient_id = gr.Number(label="Patient ID")
+        report_text = gr.Textbox(label="Report Text")
+        analyze_button = gr.Button("Analyze")
+        analyze_output = gr.Textbox(label="Output")
         
-        with gr.Tab("Patient Dashboard"):
-            dash_patient_id = gr.Number(label="Patient ID")
-            dash_password = gr.Textbox(label="Password", type="password")
-            dashboard_output = gr.Textbox(label="Patient Dashboard", interactive=False)
-            dashboard_button = gr.Button("Access Dashboard")
-            
-            dashboard_button.click(
-                fn=patient_dashboard,
-                inputs=[dash_patient_id, dash_password],
-                outputs=dashboard_output,
-            )
+        analyze_button.click(
+            fn=analyze_report,
+            inputs=[patient_id, report_text],
+            outputs=analyze_output
+        )
+    
+    with gr.Tab("Extract PDF Report"):
+        pdf_file = gr.File(label="Upload PDF")
+        extract_output = gr.Textbox(label="Extracted Text")
+        extract_button = gr.Button("Extract and Analyze")
         
-        with gr.Tab("Doctor Dashboard"):
-            doc_password = gr.Textbox(label="Doctor Password", type="password")
-            doc_dashboard_output = gr.Textbox(label="Doctor Dashboard", interactive=False)
-            doc_dashboard_button = gr.Button("Access Dashboard")
-            
-            doc_dashboard_button.click(
-                fn=doctor_dashboard,
-                inputs=doc_password,
-                outputs=doc_dashboard_output,
-            )
+        extract_button.click(
+            fn=extract_pdf_report_and_redirect,
+            inputs=pdf_file,
+            outputs=[report_text, gr.Tabs.update(selected="Analyze Medical Report")]
+        )
     
-    return app
+    with gr.Tab("Patient Dashboard"):
+        dashboard_id = gr.Number(label="Patient ID")
+        dashboard_password = gr.Textbox(label="Password", type="password")
+        dashboard_output = gr.Textbox(label="Dashboard Info")
+        dashboard_button = gr.Button("Access")
+        
+        dashboard_button.click(
+            fn=patient_dashboard,
+            inputs=[dashboard_id, dashboard_password],
+            outputs=dashboard_output
+        )
 
-app_layout().launch(share=True)
+app.launch(share=True)