Update app.py

app.py

@@ -158,154 +158,47 @@ def process_final_report(agent, file, chatbot_state: List[Dict[str, str]]) -> Tu
 
 def create_ui(agent):
     with gr.Blocks(css="""
-        :root {
-            --primary-color: #2563eb;
-            --secondary-color: #1e40af;
-            --bg-color: #f8fafc;
-            --text-color: #1e293b;
-            --user-bubble: #ffffff;
-            --bot-bubble: #f1f5f9;
-            --border-color: #e2e8f0;
-        }
         body {
+            background: #10141f;
+            color: #ffffff;
             font-family: 'Inter', sans-serif;
-            background-color: var(--bg-color);
-            color: var(--text-color);
         }
         .gradio-container {
-            max-width: 800px;
-            margin: 0 auto;
-            padding: 20px;
-            background-color: var(--bg-color);
-        }
-        .chat-container {
-            display: flex;
-            flex-direction: column;
-            height: 80vh;
-            border: 1px solid var(--border-color);
+            padding: 30px;
+            max-width: 900px;
+            margin: auto;
+            border-radius: 16px;
+            background-color: #1a1f2e;
+            box-shadow: 0 0 20px rgba(0, 0, 0, 0.5);
+        }
+        .chatbot {
+            background-color: #131720;
             border-radius: 12px;
-            overflow: hidden;
-            background-color: white;
-        }
-        .chat-header {
-            padding: 16px;
-            background-color: white;
-            border-bottom: 1px solid var(--border-color);
-            font-weight: 600;
-            font-size: 18px;
-        }
-        .chat-messages {
-            flex: 1;
-            padding: 16px;
+            padding: 20px;
+            height: 600px;
             overflow-y: auto;
-            background-color: var(--bg-color);
+            border: 1px solid #2c3344;
         }
-        .message {
-            display: flex;
-            margin-bottom: 16px;
-            align-items: flex-start;
-        }
-        .message-avatar {
-            width: 36px;
-            height: 36px;
-            border-radius: 50%;
-            margin-right: 12px;
-            background-color: var(--primary-color);
-            color: white;
-            display: flex;
-            align-items: center;
-            justify-content: center;
-            font-weight: bold;
-        }
-        .message-content {
-            max-width: 80%;
-        }
-        .user-message .message-content {
-            margin-left: auto;
-            background-color: var(--user-bubble);
-            padding: 12px 16px;
-            border-radius: 18px 18px 0 18px;
-            box-shadow: 0 1px 2px rgba(0,0,0,0.1);
-        }
-        .bot-message .message-content {
-            background-color: var(--bot-bubble);
-            padding: 12px 16px;
-            border-radius: 18px 18px 18px 0;
-            box-shadow: 0 1px 2px rgba(0,0,0,0.1);
-        }
-        .message-time {
-            font-size: 12px;
-            color: #64748b;
-            margin-top: 4px;
-        }
-        .chat-input-container {
-            padding: 16px;
-            background-color: white;
-            border-top: 1px solid var(--border-color);
-        }
-        .file-upload {
-            display: flex;
-            gap: 8px;
-            margin-bottom: 12px;
-        }
-        .upload-btn {
-            flex: 1;
-        }
-        .analyze-btn {
-            background-color: var(--primary-color);
+        .gr-button {
+            background: linear-gradient(135deg, #4b4ced, #37b6e9);
             color: white;
+            font-weight: 500;
             border: none;
-            padding: 10px 16px;
+            padding: 10px 20px;
             border-radius: 8px;
-            cursor: pointer;
-            font-weight: 500;
+            transition: background 0.3s ease;
         }
-        .analyze-btn:hover {
-            background-color: var(--secondary-color);
-        }
-        .report-download {
-            margin-top: 12px;
-            padding: 12px;
-            background-color: var(--bot-bubble);
-            border-radius: 8px;
-            display: none;
+        .gr-button:hover {
+            background: linear-gradient(135deg, #37b6e9, #4b4ced);
         }
     """) as demo:
-        with gr.Column(elem_classes="chat-container"):
-            gr.Markdown("""<div class="chat-header">Patient History AI Assistant</div>""")
-            
-            chatbot = gr.Chatbot(
-                elem_classes="chat-messages",
-                label=None,
-                show_label=False,
-                bubble_full_width=False,
-                avatar_images=[
-                    "https://ui-avatars.com/api/?name=AI&background=2563eb&color=fff&size=128",  # Bot avatar
-                    None  # User avatar (default)
-                ]
-            )
-            
-            with gr.Column(elem_classes="chat-input-container"):
-                with gr.Row(elem_classes="file-upload"):
-                    file_upload = gr.File(
-                        label="Upload Excel File",
-                        file_types=[".xlsx"],
-                        elem_classes="upload-btn",
-                        scale=4
-                    )
-                    analyze_btn = gr.Button(
-                        "Analyze",
-                        elem_classes="analyze-btn",
-                        scale=1
-                    )
-                
-                report_output = gr.File(
-                    label="Download Report",
-                    visible=False,
-                    interactive=False,
-                    elem_classes="report-download"
-                )
-
+        gr.Markdown("""# 🧠 Clinical Reasoning Assistant
+Upload clinical Excel records below and click **Analyze** to generate a medical summary.
+""")
+        chatbot = gr.Chatbot(elem_classes="chatbot")
+        file_upload = gr.File(label="Upload Excel File", file_types=[".xlsx"])
+        analyze_btn = gr.Button("Analyze")
+        report_output = gr.File(label="Download Report", visible=False)
         chatbot_state = gr.State(value=[])
 
         def update_ui(file, current_state):
@@ -327,4 +220,4 @@ if __name__ == "__main__":
         demo.launch(server_name="0.0.0.0", server_port=7860, allowed_paths=["/data/hf_cache/reports"], share=False)
     except Exception as e:
         print(f"Error: {str(e)}")
-        sys.exit(1)
+        sys.exit(1)