Update app.py

app.py

@@ -43,11 +43,14 @@ os.environ.update({
     "TOKENIZERS_PARALLELISM": "false",
     "CUDA_LAUNCH_BLOCKING": "1"
 })
+
+# Add src path for txagent
 current_dir = os.path.dirname(os.path.abspath(__file__))
 src_path = os.path.abspath(os.path.join(current_dir, "src"))
 sys.path.insert(0, src_path)
 
 from txagent.txagent import TxAgent
+
 # ==================== CORE COMPONENTS ====================
 class FileProcessor:
     """Handles all file processing operations"""
@@ -294,7 +297,7 @@ class ClinicalAgent:
                     full_response += cleaned + " "
                     yield {"role": "assistant", "content": full_response}
     
-    def analyze_records(self, message: str, history: List[Dict], files: List) -> Generator[Dict[str, Any], None, None]:
+    def analyze_records(self, message: str, history: List[Dict], files: List) -> Generator[tuple, None, None]:
         """Main analysis workflow"""
         outputs = {
             "chatbot": history.copy(),
@@ -302,13 +305,13 @@ class ClinicalAgent:
             "final_summary": "",
             "progress": {"value": "Initializing...", "visible": True}
         }
-        yield outputs
+        yield (outputs["chatbot"], outputs["download_output"], outputs["final_summary"], outputs["progress"])
         
         try:
             # Add user message
             history.append({"role": "user", "content": message})
             outputs["chatbot"] = history
-            yield outputs
+            yield (outputs["chatbot"], outputs["download_output"], outputs["final_summary"], outputs["progress"])
             
             # Process files
             extracted = []
@@ -329,7 +332,7 @@ class ClinicalAgent:
                         try:
                             extracted.extend(future.result())
                             outputs["progress"] = self._format_progress(i, len(files), "Processing files")
-                            yield outputs
+                            yield (outputs["chatbot"], outputs["download_output"], outputs["final_summary"], outputs["progress"])
                         except Exception as e:
                             logger.error(f"File processing failed: {e}")
                             extracted.append({"error": str(e)})
@@ -342,7 +345,7 @@ class ClinicalAgent:
                     "chatbot": history,
                     "progress": self._format_progress(len(files), len(files), "Files processed")
                 })
-                yield outputs
+                yield (outputs["chatbot"], outputs["download_output"], outputs["final_summary"], outputs["progress"])
 
             # Analyze content
             text_content = "\n".join(json.dumps(item) for item in extracted)
@@ -365,7 +368,7 @@ Document Excerpt (Part {idx}/{len(chunks)}):
                     "chatbot": history,
                     "progress": self._format_progress(idx, len(chunks), "Analyzing")
                 })
-                yield outputs
+                yield (outputs["chatbot"], outputs["download_output"], outputs["final_summary"], outputs["progress"])
                 
                 # Stream analysis
                 chunk_response = ""
@@ -376,7 +379,7 @@ Document Excerpt (Part {idx}/{len(chunks)}):
                         "chatbot": history,
                         "progress": self._format_progress(idx, len(chunks), "Analyzing")
                     })
-                    yield outputs
+                    yield (outputs["chatbot"], outputs["download_output"], outputs["final_summary"], outputs["progress"])
                 
                 full_analysis += f"--- Analysis Part {idx} ---\n{chunk_response}\n"
                 torch.cuda.empty_cache()
@@ -395,7 +398,7 @@ Document Excerpt (Part {idx}/{len(chunks)}):
                 "final_summary": summary,
                 "progress": {"visible": False}
             })
-            yield outputs
+            yield (outputs["chatbot"], outputs["download_output"], outputs["final_summary"], outputs["progress"])
 
         except Exception as e:
             logger.error(f"Analysis failed: {e}")
@@ -405,7 +408,7 @@ Document Excerpt (Part {idx}/{len(chunks)}):
                 "final_summary": f"Error: {str(e)}",
                 "progress": {"visible": False}
             })
-            yield outputs
+            yield (outputs["chatbot"], outputs["download_output"], outputs["final_summary"], outputs["progress"])
     
     def _format_progress(self, current: int, total: int, stage: str = "") -> Dict[str, Any]:
         """Format progress update for UI"""
@@ -414,6 +417,315 @@ Document Excerpt (Part {idx}/{len(chunks)}):
 
     def create_interface(self) -> gr.Blocks:
         """Build the Gradio interface"""
+        css = """
+        /* ==================== BASE STYLES ==================== */
+        :root {
+            --primary-color: #4f46e5;
+            --primary-dark: #4338ca;
+            --border-radius: 8px;
+            --transition: all 0.3s ease;
+            --shadow: 0 4px 12px rgba(0,0,0,0.1);
+            --font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
+            --background: #ffffff;
+            --text-color: #1e293b;
+            --chat-bg: #f8fafc;
+            --message-bg: #e2e8f0;
+            --panel-bg: rgba(248, 250, 252, 0.9);
+            --panel-dark-bg: rgba(30, 41, 59, 0.9);
+        }
+
+        [data-theme="dark"] {
+            --background: #1e2a44;
+            --text-color: #f1f5f9;
+            --chat-bg: #2d3b55;
+            --message-bg: #475569;
+            --panel-bg: var(--panel-dark-bg);
+        }
+
+        body, .gradio-container {
+            font-family: var(--font-family);
+            background: var(--background);
+            color: var(--text-color);
+            margin: 0;
+            padding: 0;
+            transition: var(--transition);
+        }
+
+        /* ==================== LAYOUT ==================== */
+        .gradio-container {
+            max-width: 1200px;
+            margin: 0 auto;
+            padding: 1.5rem;
+            display: flex;
+            flex-direction: column;
+            gap: 1.5rem;
+        }
+
+        .chat-container {
+            background: var(--chat-bg);
+            border-radius: var(--border-radius);
+            border: 1px solid #e2e8f0;
+            padding: 1.5rem;
+            min-height: 50vh;
+            max-height: 80vh;
+            overflow-y: auto;
+            box-shadow: var(--shadow);
+            margin-bottom: 4rem;
+        }
+
+        .summary-panel {
+            background: var(--panel-bg);
+            border-left: 4px solid var(--primary-color);
+            padding: 1rem;
+            border-radius: var(--border-radius);
+            margin-bottom: 1rem;
+            box-shadow: var(--shadow);
+            backdrop-filter: blur(8px);
+        }
+
+        .upload-area {
+            border: 2px dashed #cbd5e1;
+            border-radius: var(--border-radius);
+            padding: 1.5rem;
+            margin: 0.75rem 0;
+            transition: var(--transition);
+        }
+
+        .upload-area:hover {
+            border-color: var(--primary-color);
+            background: rgba(79, 70, 229, 0.05);
+        }
+
+        /* ==================== COMPONENTS ==================== */
+        .chat__message {
+            margin: 0.75rem 0;
+            padding: 0.75rem 1rem;
+            border-radius: var(--border-radius);
+            max-width: 85%;
+            transition: var(--transition);
+            background: var(--message-bg);
+            border: 1px solid rgba(0,0,0,0.05);
+            animation: messageFade 0.3s ease;
+        }
+
+        .chat__message:hover {
+            transform: translateY(-2px);
+            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        }
+
+        .chat__message.user {
+            background: linear-gradient(135deg, var(--primary-color), var(--primary-dark));
+            color: white;
+            margin-left: auto;
+        }
+
+        .chat__message.assistant {
+            background: var(--message-bg);
+            color: var(--text-color);
+        }
+
+        .input-container {
+            display: flex;
+            align-items: center;
+            gap: 0.75rem;
+            background: var(--chat-bg);
+            padding: 0.75rem 1rem;
+            border-radius: 1.5rem;
+            box-shadow: var(--shadow);
+            position: sticky;
+            bottom: 1rem;
+            z-index: 10;
+        }
+
+        .input__textbox {
+            flex-grow: 1;
+            border: none;
+            background: transparent;
+            color: var(--text-color);
+            outline: none;
+            font-size: 1rem;
+        }
+
+        .input__textbox:focus {
+            border-bottom: 2px solid var(--primary-color);
+        }
+
+        .submit-btn {
+            background: linear-gradient(135deg, var(--primary-color), var(--primary-dark));
+            color: white;
+            border: none;
+            border-radius: 1rem;
+            padding: 0.5rem 1.25rem;
+            font-size: 0.9rem;
+            transition: var(--transition);
+        }
+
+        .submit-btn:hover {
+            transform: scale(1.05);
+        }
+
+        .submit-btn:active {
+            animation: glow 0.3s ease;
+        }
+
+        .tooltip {
+            position: relative;
+        }
+
+        .tooltip:hover::after {
+            content: attr(data-tip);
+            position: absolute;
+            top: -2.5rem;
+            left: 50%;
+            transform: translateX(-50%);
+            background: #1e293b;
+            color: white;
+            padding: 0.4rem 0.8rem;
+            border-radius: 0.4rem;
+            font-size: 0.85rem;
+            max-width: 200px;
+            white-space: normal;
+            text-align: center;
+            z-index: 1000;
+            animation: fadeIn 0.3s ease;
+        }
+
+        .progress-tracker {
+            position: relative;
+            padding: 0.5rem;
+            background: var(--message-bg);
+            border-radius: var(--border-radius);
+            margin-top: 0.75rem;
+            overflow: hidden;
+        }
+
+        .progress-tracker::before {
+            content: '';
+            position: absolute;
+            top: 0;
+            left: 0;
+            height: 100%;
+            width: 0;
+            background: linear-gradient(to right, var(--primary-color), var(--primary-dark));
+            opacity: 0.3;
+            animation: progress 2s ease-in-out infinite;
+        }
+
+        /* ==================== ANIMATIONS ==================== */
+        @keyframes glow {
+            0%, 100% { transform: scale(1); opacity: 1; }
+            50% { transform: scale(1.1); opacity: 0.8; }
+        }
+
+        @keyframes fadeIn {
+            from { opacity: 0; }
+            to { opacity: 1; }
+        }
+
+        @keyframes messageFade {
+            from { opacity: 0; transform: translateY(10px) scale(0.95); }
+            to { opacity: 1; transform: translateY(0) scale(1); }
+        }
+
+        @keyframes progress {
+            0% { width: 0; }
+            50% { width: 60%; }
+            100% { width: 0; }
+        }
+
+        /* ==================== THEMES ==================== */
+        [data-theme="dark"] .chat-container {
+            border-color: #475569;
+        }
+
+        [data-theme="dark"] .upload-area {
+            border-color: #64748b;
+        }
+
+        [data-theme="dark"] .upload-area:hover {
+            background: rgba(79, 70, 229, 0.1);
+        }
+
+        [data-theme="dark"] .summary-panel {
+            border-left-color: #818cf8;
+        }
+
+        /* ==================== MEDIA QUERIES ==================== */
+        @media (max-width: 768px) {
+            .gradio-container {
+                padding: 1rem;
+            }
+
+            .chat-container {
+                min-height: 40vh;
+                max-height: 70vh;
+                margin-bottom: 3.5rem;
+            }
+
+            .summary-panel {
+                padding: 0.75rem;
+            }
+
+            .upload-area {
+                padding: 1rem;
+            }
+
+            .input-container {
+                gap: 0.5rem;
+                padding: 0.5rem;
+            }
+
+            .submit-btn {
+                padding: 0.4rem 1rem;
+            }
+        }
+
+        @media (max-width: 480px) {
+            .chat-container {
+                padding: 1rem;
+                margin-bottom: 3rem;
+            }
+
+            .input-container {
+                flex-direction: column;
+                padding: 0.5rem;
+            }
+
+            .input__textbox {
+                font-size: 0.9rem;
+            }
+
+            .submit-btn {
+                width: 100%;
+                padding: 0.5rem;
+                font-size: 0.85rem;
+            }
+
+            .chat__message {
+                max-width: 90%;
+                padding: 0.5rem 0.75rem;
+            }
+
+            .tooltip:hover::after {
+                top: auto;
+                bottom: -2.5rem;
+                max-width: 80vw;
+            }
+        }
+        """
+        
+        js = """
+        function applyTheme(theme) {
+            document.documentElement.setAttribute('data-theme', theme);
+            localStorage.setItem('theme', theme);
+        }
+
+        document.addEventListener('DOMContentLoaded', () => {
+            const savedTheme = localStorage.getItem('theme') || 'light';
+            applyTheme(savedTheme);
+        });
+        """
+
         with gr.Blocks(
             theme=gr.themes.Soft(
                 primary_hue="indigo",
@@ -421,30 +733,13 @@ Document Excerpt (Part {idx}/{len(chunks)}):
                 neutral_hue="slate"
             ),
             title="Clinical Oversight Assistant",
-            css="""
-            .summary-panel {
-                border-left: 4px solid #4f46e5;
-                padding: 16px;
-                background: #f8fafc;
-                border-radius: 8px;
-                margin-bottom: 16px;
-            }
-            .upload-area {
-                border: 2px dashed #cbd5e1;
-                border-radius: 8px;
-                padding: 24px;
-                margin: 12px 0;
-            }
-            .chat-container {
-                border-radius: 8px;
-                border: 1px solid #e2e8f0;
-            }
-            """
+            css=css,
+            js=js
         ) as app:
             # Header
             gr.Markdown("""
             <div style='text-align: center; margin-bottom: 24px;'>
-                <h1 style='color: #4f46e5; margin-bottom: 8px;'>🩺 Clinical Oversight Assistant</h1>
+                <h1 style='color: var(--primary-color); margin-bottom: 8px;'>🩺 Clinical Oversight Assistant</h1>
                 <p style='color: #64748b;'>
                     AI-powered analysis for identifying potential missed diagnoses in patient records
                 </p>
@@ -454,7 +749,9 @@ Document Excerpt (Part {idx}/{len(chunks)}):
             with gr.Row(equal_height=False):
                 # Main Chat Panel
                 with gr.Column(scale=3):
-                    gr.Markdown("**Clinical Analysis Conversation**")
+                    gr.Markdown(
+                        "<div class='tooltip' data-tip='View conversation history'>**Clinical Analysis Conversation**</div>"
+                    )
                     chatbot = gr.Chatbot(
                         label="",
                         height=650,
@@ -471,14 +768,18 @@ Document Excerpt (Part {idx}/{len(chunks)}):
                 # Results Panel
                 with gr.Column(scale=1):
                     with gr.Group():
-                        gr.Markdown("**Clinical Summary**")
+                        gr.Markdown(
+                            "<div class='tooltip' data-tip='Summary of findings'>**Clinical Summary**</div>"
+                        )
                         final_summary = gr.Markdown(
-                            "Analysis results will appear here...",
+                            "<div class='tooltip' data-tip='Analysis results'>Analysis results will appear here...</div>",
                             elem_classes=["summary-panel"]
                         )
                     
                     with gr.Group():
-                        gr.Markdown("**Report Export**")
+                        gr.Markdown(
+                            "<div class='tooltip' data-tip='Download report'>**Report Export**</div>"
+                        )
                         download_output = gr.File(
                             label="Download Full Analysis",
                             visible=False,
@@ -491,29 +792,36 @@ Document Excerpt (Part {idx}/{len(chunks)}):
                     file_types=[".pdf", ".csv", ".xls", ".xlsx"],
                     file_count="multiple",
                     label="Upload Patient Records",
-                    elem_classes=["upload-area"]
+                    elem_classes=["upload-area"],
+                    elem_id="file-upload"
                 )
 
-            with gr.Row():
+            with gr.Row(elem_classes=["input-container"]):
                 user_input = gr.Textbox(
                     placeholder="Enter your clinical query or analysis request...",
                     show_label=False,
                     container=False,
                     scale=7,
-                    autofocus=True
+                    autofocus=True,
+                    elem_classes=["input__textbox"],
+                    elem_id="user-input"
                 )
                 submit_btn = gr.Button(
                     "Analyze",
                     variant="primary",
                     scale=1,
-                    min_width=120
+                    min_width=120,
+                    elem_classes=["submit-btn"],
+                    elem_id="submit-btn"
                 )
 
             # Hidden progress tracker
             progress_tracker = gr.Textbox(
                 label="Analysis Progress",
                 visible=False,
-                interactive=False
+                interactive=False,
+                elem_classes=["progress-tracker"],
+                elem_id="progress-tracker"
             )
 
             # Event handlers
@@ -532,7 +840,7 @@ Document Excerpt (Part {idx}/{len(chunks)}):
             )
             
             app.load(
-                lambda: [[], None, "", "", None, {"visible": False}],
+                lambda: [[], None, "<div class='tooltip' data-tip='Analysis results'>Analysis results will appear here...</div>", "", None, {"visible": False}],
                 outputs=[chatbot, download_output, final_summary, user_input, file_upload, progress_tracker],
                 queue=False
             )