Update app.py

app.py

@@ -53,6 +53,9 @@ sys.path.insert(0, src_path)
 
 from txagent.txagent import TxAgent
 
+# Log Gradio version for debugging
+logger.info(f"Gradio version: {gr.__version__}")
+
 # ==================== UTILITY FUNCTIONS ====================
 def sanitize_text(text: str) -> str:
     """Clean and sanitize text input"""
@@ -303,9 +306,11 @@ class ClinicalOversightApp:
 
     def cleanup_resources(self):
         """Clean up GPU memory and collect garbage"""
+        logger.info("Cleaning up resources...")
         torch.cuda.empty_cache()
         gc.collect()
         if torch.distributed.is_initialized():
+            logger.info("Destroying PyTorch distributed process group...")
             torch.distributed.destroy_process_group()
 
     def process_response_stream(self, prompt: str, history: List[dict]) -> Generator[dict, None, None]:
@@ -453,6 +458,7 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
             background: var(--background);
             color: var(--text-color);
+            transition: all 0.4s ease;
         }
         .gradio-container {
             max-width: 800px;
@@ -514,7 +520,9 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             color: var(--text-color);
             outline: none;
             font-size: 1em;
-            placeholder-opacity: 0.6;
+        }
+        .input-textbox:focus {
+            border-bottom: 2px solid #007bff;
         }
         .send-btn {
             background: linear-gradient(135deg, #007bff, #0056b3);
@@ -529,7 +537,7 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             transform: scale(1.05);
         }
         .send-btn:active {
-            animation: pulse 0.3s ease;
+            animation: glow 0.3s ease;
         }
         .sidebar {
             background: var(--sidebar-bg);
@@ -544,6 +552,8 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             top: 100px;
             width: 300px;
             z-index: 1000;
+            backdrop-filter: blur(10px);
+            background: rgba(241, 243, 245, 0.8);
         }
         .sidebar-hidden {
             transform: translateX(100%);
@@ -578,13 +588,31 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             align-items: center;
             gap: 8px;
         }
+        .tooltip {
+            position: relative;
+        }
+        .tooltip:hover::after {
+            content: attr(data-tooltip);
+            position: absolute;
+            bottom: 100%;
+            left: 50%;
+            transform: translateX(-50%);
+            background: #333;
+            color: white;
+            padding: 6px 12px;
+            border-radius: 6px;
+            font-size: 0.85em;
+            white-space: nowrap;
+            z-index: 1000;
+            animation: fadeIn 0.3s ease;
+        }
         .loading-spinner {
             position: absolute;
             bottom: 80px;
             left: 50%;
             transform: translateX(-50%);
             font-size: 1.2em;
-            animation: pulse 1.5s ease infinite;
+            animation: glow 1.5s ease infinite;
         }
         .typing-indicator {
             display: none;
@@ -597,20 +625,46 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
             display: block;
             animation: blink 1s step-end infinite;
         }
-        @keyframes pulse {
-            0%, 100% { transform: translateX(-50%) scale(1); opacity: 1; }
-            50% { transform: translateX(-50%) scale(1.2); opacity: 0.7; }
+        .progress-text {
+            position: relative;
+            padding: 8px;
+            background: var(--message-bg);
+            border-radius: 8px;
+            margin-top: 12px;
+        }
+        .progress-text::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            left: 0;
+            height: 100%;
+            width: 0;
+            background: #007bff;
+            opacity: 0.2;
+            animation: progress 2s linear infinite;
+        }
+        @keyframes glow {
+            0%, 100% { transform: translateX(-50%) scale(1); opacity: 1; color: #007bff; }
+            50% { transform: translateX(-50%) scale(1.2); opacity: 0.7; color: #0056b3; }
         }
         @keyframes blink {
             50% { opacity: 0.3; }
         }
+        @keyframes fadeIn {
+            from { opacity: 0; }
+            to { opacity: 1; }
+        }
+        @keyframes progress {
+            0% { width: 0; }
+            50% { width: 50%; }
+            100% { width: 0; }
+        }
         :root {
             --background: #ffffff;
             --text-color: #333333;
             --chat-bg: #f9fafb;
             --message-bg: #e5e5ea;
             --sidebar-bg: #f1f3f5;
-            transition: background 0.3s ease, color 0.3s ease;
         }
         [data-theme="dark"] {
             --background: #1e2a44;
@@ -655,6 +709,13 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
         function toggleSidebar() {
             const sidebar = document.querySelector('.sidebar');
             sidebar.classList.toggle('sidebar-hidden');
+            if (!sidebar.classList.contains('sidebar-hidden')) {
+                setTimeout(() => {
+                    if (window.innerWidth <= 600) {
+                        sidebar.classList.add('sidebar-hidden');
+                    }
+                }, 5000);
+            }
         }
 
         document.addEventListener('DOMContentLoaded', () => {
@@ -665,165 +726,170 @@ Patient Record (Chunk {chunk_idx}/{len(chunks)}):
         """
 
         with gr.Blocks(theme=gr.themes.Default(), css=css, js=js, title="Clinical Oversight Assistant") as app:
-            theme_state = gr.State(value="light")
-            sidebar_state = gr.State(value=False)
-            
-            gr.HTML("""
-            <div class='header'>
-                <h1 style='color: var(--text-color);'>🩺 Clinical Oversight Assistant</h1>
-                <p style='color: var(--text-color); opacity: 0.7;'>
-                    AI-powered analysis of patient records for missed diagnoses
-                </p>
-            </div>
-            <div class='sidebar-backdrop'></div>
-            """)
-            
-            theme_button = gr.Button("🌙 Dark Mode", elem_classes="theme-toggle")
-            
-            with gr.Column(elem_classes="chat-container"):
-                chatbot = gr.Chatbot(
-                    label="Clinical Analysis",
-                    height="100%",
-                    show_copy_button=True,
-                    type="messages",
-                    elem_classes="chatbot",
-                    render_markdown=True
+            try:
+                theme_state = gr.State(value="light")
+                sidebar_state = gr.State(value=False)
+                
+                gr.HTML("""
+                <div class='header'>
+                    <h1 style='color: var(--text-color);'>🩺 Clinical Oversight Assistant</h1>
+                    <p style='color: var(--text-color); opacity: 0.7;'>
+                        AI-powered analysis of patient records for missed diagnoses
+                    </p>
+                </div>
+                <div class='sidebar-backdrop'></div>
+                """)
+                
+                theme_button = gr.Button("🌙 Dark Mode", elem_classes="theme-toggle")
+                
+                with gr.Column(elem_classes="chat-container"):
+                    chatbot = gr.Chatbot(
+                        label="Clinical Analysis",
+                        height="100%",
+                        show_copy_button=True,
+                        type="messages",
+                        elem_classes="chatbot",
+                        render_markdown=True
+                    )
+                    gr.HTML("<div class='loading-spinner' style='display: none;'>⏳</div>")
+                    gr.HTML("<div class='typing-indicator'>Typing...</div>")
+                
+                with gr.Row():
+                    tools_button = gr.Button("📂 Tools", variant="secondary")
+                
+                with gr.Column(elem_classes="sidebar"):
+                    gr.Markdown("### 📎 Upload Records", elem_classes="tooltip", data_tooltip="Upload patient records")
+                    file_upload = gr.File(
+                        file_types=[".pdf", ".csv", ".xls", ".xlsx"],
+                        file_count="multiple",
+                        label="Patient Records",
+                        elem_classes="tooltip",
+                        data_tooltip="Select PDF, CSV, or Excel files"
+                    )
+                    gr.Markdown("### 📝 Analysis Summary", elem_classes="tooltip", data_tooltip="Summary of findings")
+                    final_summary = gr.Markdown(
+                        "Analysis results will appear here...",
+                        elem_classes="tooltip",
+                        data_tooltip="View analysis results"
+                    )
+                    gr.Markdown("### 📄 Full Report", elem_classes="tooltip", data_tooltip="Download full report")
+                    download_output = gr.File(
+                        label="Download Report",
+                        visible=False,
+                        interactive=False,
+                        elem_classes="tooltip",
+                        data_tooltip="Download analysis report"
+                    )
+                
+                with gr.Row(elem_classes="input-container"):
+                    msg_input = gr.Textbox(
+                        placeholder="Ask about potential oversights or upload files...",
+                        show_label=False,
+                        container=False,
+                        elem_classes="input-textbox",
+                        autofocus=True
+                    )
+                    send_btn = gr.Button(
+                        "Analyze",
+                        variant="primary",
+                        elem_classes="send-btn"
+                    )
+                
+                progress_text = gr.Textbox(
+                    label="Progress Status",
+                    visible=False,
+                    interactive=False,
+                    elem_classes="progress-text"
                 )
-                gr.HTML("<div class='loading-spinner'>⏳</div>")
-                gr.HTML("<div class='typing-indicator'>Typing...</div>")
-            
-            with gr.Row():
-                tools_button = gr.Button("📂 Tools", variant="secondary")
-            
-            with gr.Column(elem_classes="sidebar"):
-                gr.Markdown("### 📎 Upload Records", elem_classes="tooltip", elem_id="upload-tooltip")
-                file_upload = gr.File(
-                    file_types=[".pdf", ".csv", ".xls", ".xlsx"],
-                    file_count="multiple",
-                    label="Patient Records",
-                    elem_classes="tooltip",
-                    elem_id="upload-input"
+
+                def show_loading(state: bool) -> dict:
+                    return {
+                        "value": "<div class='loading-spinner'>⏳</div>" if state else "<div class='loading-spinner' style='display: none;'>⏳</div>",
+                        "visible": state
+                    }
+
+                def show_typing(state: bool) -> dict:
+                    return {
+                        "value": f"<div class='typing-indicator{' active' if state else ''}'>Typing...</div>",
+                        "visible": state
+                    }
+
+                # Theme toggle handler
+                theme_button.click(
+                    fn=self.toggle_theme,
+                    inputs=[theme_state],
+                    outputs=[theme_state, theme_button],
+                    _js="function(theme) { applyTheme(theme); }"
                 )
-                gr.Markdown("### 📝 Analysis Summary", elem_classes="tooltip", elem_id="summary-tooltip")
-                final_summary = gr.Markdown(
-                    "Analysis results will appear here...",
-                    elem_classes="tooltip",
-                    elem_id="summary-output"
+
+                # Sidebar toggle handler
+                tools_button.click(
+                    fn=self.toggle_sidebar,
+                    inputs=[sidebar_state],
+                    outputs=[sidebar_state],
+                    _js="toggleSidebar"
                 )
-                gr.Markdown("### 📄 Full Report", elem_classes="tooltip", elem_id="report-tooltip")
-                download_output = gr.File(
-                    label="Download Report",
-                    visible=False,
-                    interactive=False,
-                    elem_classes="tooltip",
-                    elem_id="download-output"
+
+                # Analysis handlers
+                send_btn.click(
+                    fn=show_loading,
+                    inputs=[gr.State(value=True)],
+                    outputs=[chatbot]
+                ).then(
+                    fn=show_typing,
+                    inputs=[gr.State(value=True)],
+                    outputs=[chatbot]
+                ).then(
+                    fn=self.analyze,
+                    inputs=[msg_input, chatbot, file_upload],
+                    outputs=[chatbot, download_output, final_summary, progress_text],
+                    show_progress="hidden"
+                ).then(
+                    fn=show_loading,
+                    inputs=[gr.State(value=False)],
+                    outputs=[chatbot]
+                ).then(
+                    fn=show_typing,
+                    inputs=[gr.State(value=False)],
+                    outputs=[chatbot]
                 )
-            
-            with gr.Row(elem_classes="input-container"):
-                msg_input = gr.Textbox(
-                    placeholder="Ask about potential oversights or upload files...",
-                    show_label=False,
-                    container=False,
-                    elem_classes="input-textbox",
-                    autofocus=True
+                
+                msg_input.submit(
+                    fn=show_loading,
+                    inputs=[gr.State(value=True)],
+                    outputs=[chatbot]
+                ).then(
+                    fn=show_typing,
+                    inputs=[gr.State(value=True)],
+                    outputs=[chatbot]
+                ).then(
+                    fn=self.analyze,
+                    inputs=[msg_input, chatbot, file_upload],
+                    outputs=[chatbot, download_output, final_summary, progress_text],
+                    show_progress="hidden"
+                ).then(
+                    fn=show_loading,
+                    inputs=[gr.State(value=False)],
+                    outputs=[chatbot]
+                ).then(
+                    fn=show_typing,
+                    inputs=[gr.State(value=False)],
+                    outputs=[chatbot]
                 )
-                send_btn = gr.Button(
-                    "Analyze",
-                    variant="primary",
-                    elem_classes="send-btn"
+                
+                app.load(
+                    fn=lambda: [
+                        [], None, "", "", None, {"visible": False}, "light", False, "🌙 Dark Mode"
+                    ],
+                    outputs=[chatbot, download_output, final_summary, msg_input, file_upload, progress_text, theme_state, sidebar_state, theme_button],
+                    queue=False
                 )
-            
-            progress_text = gr.Textbox(
-                label="Progress Status",
-                visible=False,
-                interactive=False,
-                elem_classes="progress-text"
-            )
-
-            def show_loading(state: bool) -> dict:
-                return {
-                    "value": "<div class='loading-spinner'>⏳</div>" if state else "<div class='loading-spinner' style='display: none;'>⏳</div>",
-                    "visible": state
-                }
-
-            def show_typing(state: bool) -> dict:
-                return {
-                    "value": f"<div class='typing-indicator{' active' if state else ''}'>Typing...</div>",
-                    "visible": state
-                }
-
-            # Theme toggle handler
-            theme_button.click(
-                self.toggle_theme,
-                inputs=[theme_state],
-                outputs=[theme_state, theme_button],
-                _js="theme => applyTheme(theme)"
-            )
-
-            # Sidebar toggle handler
-            tools_button.click(
-                self.toggle_sidebar,
-                inputs=[sidebar_state],
-                outputs=[sidebar_state],
-                _js="() => toggleSidebar()"
-            )
-
-            # Analysis handlers
-            send_btn.click(
-                show_loading,
-                inputs=[gr.State(value=True)],
-                outputs=[chatbot]
-            ).then(
-                show_typing,
-                inputs=[gr.State(value=True)],
-                outputs=[chatbot]
-            ).then(
-                self.analyze,
-                inputs=[msg_input, chatbot, file_upload],
-                outputs=[chatbot, download_output, final_summary, progress_text],
-                show_progress="hidden"
-            ).then(
-                show_loading,
-                inputs=[gr.State(value=False)],
-                outputs=[chatbot]
-            ).then(
-                show_typing,
-                inputs=[gr.State(value=False)],
-                outputs=[chatbot]
-            )
-            
-            msg_input.submit(
-                show_loading,
-                inputs=[gr.State(value=True)],
-                outputs=[chatbot]
-            ).then(
-                show_typing,
-                inputs=[gr.State(value=True)],
-                outputs=[chatbot]
-            ).then(
-                self.analyze,
-                inputs=[msg_input, chatbot, file_upload],
-                outputs=[chatbot, download_output, final_summary, progress_text],
-                show_progress="hidden"
-            ).then(
-                show_loading,
-                inputs=[gr.State(value=False)],
-                outputs=[chatbot]
-            ).then(
-                show_typing,
-                inputs=[gr.State(value=False)],
-                outputs=[chatbot]
-            )
-            
-            app.load(
-                lambda: [
-                    [], None, "", "", None, {"visible": False}, "light", False, "🌙 Dark Mode"
-                ],
-                outputs=[chatbot, download_output, final_summary, msg_input, file_upload, progress_text, theme_state, sidebar_state, theme_button],
-                queue=False
-            )
-
-        return app
+
+            except Exception as e:
+                logger.error(f"Interface creation failed: {e}")
+                self.cleanup_resources()
+                raise
+            return app
 
 # ==================== APPLICATION ENTRY POINT ====================
 if __name__ == "__main__":