updated code ✅✅

mediSync/app.py

@@ -467,22 +467,27 @@ def create_interface():
         css=custom_css
     ) as interface:
         
-        # Appointment ID handling - simplified approach
-        appointment_id = gr.Textbox(visible=False)
-        
-        # Initialize appointment ID from URL using Gradio's built-in method
+        # Session state to track if consultation was properly recorded
+        session_state = gr.State({
+            'recorded': False,
+            'appointment_id': ''
+        })
+
+        # Initialize appointment ID from URL
         def get_appointment_id():
             try:
                 from gradio.context import Context
                 if hasattr(Context, 'request') and hasattr(Context.request, 'query_params'):
-                    return Context.request.query_params.get("appointment_id", "")
+                    appointment_id = Context.request.query_params.get("appointment_id", "")
+                    if appointment_id:
+                        return {'recorded': False, 'appointment_id': appointment_id}
             except Exception as e:
                 logger.warning(f"Could not get URL parameters: {str(e)}")
-            return ""
-        
+            return {'recorded': False, 'appointment_id': ''}
+
         interface.load(
             fn=get_appointment_id,
-            outputs=[appointment_id]
+            outputs=[session_state]
         )
 
         gr.Markdown("""
@@ -492,154 +497,87 @@ def create_interface():
         to provide comprehensive medical insights.
         """)
 
-        with gr.Tab("Multimodal Analysis"):
-            with gr.Row():
-                with gr.Column():
-                    multi_img_input = gr.Image(label="Upload X-ray Image", type="pil")
-                    multi_img_enhance = gr.Button("Enhance Image")
-                    multi_text_input = gr.Textbox(
-                        label="Enter Medical Report Text",
-                        placeholder="Enter the radiologist's report text here...",
-                        lines=10,
-                        value=example_report if not sample_image_path else None,
-                    )
-                    multi_analyze_btn = gr.Button("Analyze Image & Text", variant="primary")
-                with gr.Column():
-                    multi_results = gr.HTML(label="Analysis Results")
-                    multi_plot = gr.HTML(label="Visualization")
-
-            if sample_image_path:
-                gr.Examples(
-                    examples=[[sample_image_path, example_report]],
-                    inputs=[multi_img_input, multi_text_input],
-                    label="Example X-ray and Report",
-                )
-
-        with gr.Tab("Image Analysis"):
-            with gr.Row():
-                with gr.Column():
-                    img_input = gr.Image(label="Upload X-ray Image", type="pil")
-                    img_enhance = gr.Button("Enhance Image")
-                    img_analyze_btn = gr.Button("Analyze Image", variant="primary")
-                with gr.Column():
-                    img_output = gr.Image(label="Processed Image")
-                    img_results = gr.HTML(label="Analysis Results")
-                    img_plot = gr.HTML(label="Visualization")
-
-            if sample_image_path:
-                gr.Examples(
-                    examples=[[sample_image_path]],
-                    inputs=[img_input],
-                    label="Example X-ray Image",
-                )
-
-        with gr.Tab("Text Analysis"):
-            with gr.Row():
-                with gr.Column():
-                    text_input = gr.Textbox(
-                        label="Enter Medical Report Text",
-                        placeholder="Enter the radiologist's report text here...",
-                        lines=10,
-                        value=example_report,
-                    )
-                    text_analyze_btn = gr.Button("Analyze Text", variant="primary")
-                with gr.Column():
-                    text_output = gr.Textbox(label="Processed Text")
-                    text_results = gr.HTML(label="Analysis Results")
-                    text_plot = gr.HTML(label="Entity Visualization")
-
-            gr.Examples(
-                examples=[[example_report]],
-                inputs=[text_input],
-                label="Example Medical Report",
-            )
-
-        with gr.Tab("About"):
-            gr.Markdown("""
-            ## About MediSync
-            
-            MediSync is an AI-powered healthcare solution that uses multi-modal analysis to provide comprehensive insights from medical images and reports.
-            """)
+        # ... (keep all your existing tabs and components)
 
         with gr.Row():
             with gr.Column():
                 end_consultation_btn = gr.Button("End Consultation", variant="stop", size="lg")
                 completion_status = gr.HTML()
 
-        # Event handlers
-        multi_img_enhance.click(app.enhance_image, inputs=multi_img_input, outputs=multi_img_input)
-        multi_analyze_btn.click(
-            app.analyze_multimodal,
-            inputs=[multi_img_input, multi_text_input],
-            outputs=[multi_results, multi_plot],
-        )
-        img_enhance.click(app.enhance_image, inputs=img_input, outputs=img_output)
-        img_analyze_btn.click(
-            app.analyze_image,
-            inputs=img_input,
-            outputs=[img_output, img_results, img_plot],
-        )
-        text_analyze_btn.click(
-            app.analyze_text,
-            inputs=text_input,
-            outputs=[text_output, text_results, text_plot],
-        )
-
-        def complete_consultation(appointment_id):
-            if not appointment_id:
+        def complete_consultation(session_state):
+            """Handle consultation completion with proper session recording."""
+            if not session_state['appointment_id']:
                 error_html = """
                 <div class="alert-box alert-error">
                     <strong>No Appointment ID Found</strong>
-                    <p>Your session may not be properly recorded.</p>
+                    <p>Please contact support to ensure your session is recorded.</p>
                 </div>
                 """
                 logger.error("Completion attempted without appointment ID")
-                return error_html
+                return {'recorded': False, 'appointment_id': ''}, error_html
             
             try:
+                # First verify if already recorded
+                if session_state['recorded']:
+                    success_html = """
+                    <div class="alert-box alert-success">
+                        <strong>Consultation Already Completed</strong>
+                        <p>You will be redirected shortly.</p>
+                        <script>
+                            setTimeout(() => window.location.href = "http://127.0.0.1:600/doctors", 1000);
+                        </script>
+                    </div>
+                    """
+                    return session_state, success_html
+
+                # Call your Flask endpoint to record completion
                 response = requests.post(
                     "http://127.0.0.1:600/complete_consultation",
-                    json={"appointment_id": appointment_id},
+                    json={"appointment_id": session_state['appointment_id']},
                     timeout=10
                 )
 
                 if response.ok:
                     success_html = """
                     <div class="alert-box alert-success">
-                        <strong>Consultation Completed</strong>
+                        <strong>Consultation Successfully Recorded</strong>
+                        <p>Your session has been properly recorded.</p>
                         <p>You will be redirected shortly.</p>
                         <script>
                             setTimeout(() => window.location.href = "http://127.0.0.1:600/doctors", 2000);
                         </script>
                     </div>
                     """
-                    logger.info(f"Successfully completed appointment {appointment_id}")
-                    return success_html
+                    logger.info(f"Successfully recorded appointment {session_state['appointment_id']}")
+                    return {'recorded': True, 'appointment_id': session_state['appointment_id']}, success_html
+                
                 error_msg = f"Server error: {response.status_code}"
                 if response.text:
                     error_msg += f" - {response.text[:200]}"
                 logger.error(error_msg)
-                return f"""
+                return session_state, f"""
                 <div class="alert-box alert-error">
-                    <strong>Error Completing Appointment</strong>
+                    <strong>Recording Failed</strong>
                     <p>{error_msg}</p>
+                    <p>Please try again or contact support.</p>
                 </div>
                 """
             except Exception as e:
                 logger.error(f"Completion error: {str(e)}")
-                return f"""
+                return session_state, f"""
                 <div class="alert-box alert-error">
                     <strong>Connection Error</strong>
                     <p>{str(e)}</p>
+                    <p>Please check your connection and try again.</p>
                 </div>
                 """
 
         end_consultation_btn.click(
             fn=complete_consultation,
-            inputs=[appointment_id],
-            outputs=[completion_status]
+            inputs=[session_state],
+            outputs=[session_state, completion_status]
         )
-
+        
     try:
         interface.launch()
     except Exception as e: