Revamp app.py for the Clinical Trial Matching Chatbot, introducing a new theme and enhanced CSS for improved visual appeal. Restructure content to provide clearer guidance on the tool's functionality and workflow, while optimizing the chat interface for better user interaction.

app.py

@@ -372,37 +372,99 @@ if __name__ == "__main__":
     def chat_function(message, history):
         return get_clinical_trials(message)
     
-    # Create simple interface without any custom CSS that could cause layout issues
-    with gr.Blocks(title="Clinical Trial Matching") as demo:
+    # Create optimized chatbot with simple, reliable CSS
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:
+        # Add simple CSS for styling
+        gr.HTML("""
+        <style>
+        .workflow-card {
+            background: linear-gradient(135deg, #f0f9ff, #e0f2fe);
+            border: 2px solid #3b82f6;
+            border-radius: 12px;
+            padding: 15px;
+            margin: 8px;
+            text-align: center;
+            box-shadow: 0 3px 6px rgba(0,0,0,0.1);
+        }
+        .workflow-card-1 { background: linear-gradient(135deg, #fef3c7, #fde68a); border-color: #f59e0b; }
+        .workflow-card-2 { background: linear-gradient(135deg, #ddd6fe, #c4b5fd); border-color: #8b5cf6; }
+        .workflow-card-3 { background: linear-gradient(135deg, #bfdbfe, #93c5fd); border-color: #3b82f6; }
+        .workflow-card-4 { background: linear-gradient(135deg, #bbf7d0, #86efac); border-color: #10b981; }
+        .workflow-card-5 { background: linear-gradient(135deg, #fecaca, #fca5a5); border-color: #ef4444; }
+        </style>
+        """)
+        
         gr.Markdown("# 🏥 Clinical Trial Matching Chatbot")
-        gr.Markdown("**AI-Powered Clinical Trial Matching** - Describe a patient's condition to find matching trials quickly.")
+        gr.Markdown("**AI-Powered Clinical Trial Matching with GPU acceleration and parallel processing.** Describe a patient's condition to find matching trials quickly.")
         
-        with gr.Accordion("About This Tool", open=False):
-            gr.Markdown("""
-            This AI-powered tool helps with **Patient Recruitment & Enrollment** by finding the most relevant clinical trials based on patient cases. 
-            It uses advanced language models to analyze patient descriptions and match them with trial criteria.
-            """)
-            
-        with gr.Accordion("How It Works", open=False):
-            gr.Markdown("""
-            **Workflow:**
-            1. 📝 **Patient Summary** - Enter patient details
-            2. 🔑 **Keyword Extraction** - AI extracts medical conditions 
-            3. 🔎 **Trial Retrieval** - Search matching trials
-            4. ✅ **Eligibility Check** - Analyze inclusion/exclusion criteria
-            5. 🏆 **Results** - Rank and display top 5 matches
-            """)
+        gr.Markdown("## About the Tool")
+        gr.Markdown("""
+        This AI-powered tool is designed to support **Patient Recruitment & Enrollment** by retrieving the most relevant clinical trials based on a given patient case. It is built on top of a large language model fine-tuned on clinical trial metadata, including summaries, inclusion/exclusion criteria, drugs, and target diseases. The dataset includes several actively recruiting Pfizer trials.
+        """)
+        
+        gr.Markdown("## How It Works")
+        gr.Markdown("""
+        Users input a patient case describing clinical history and findings. The model then identifies and ranks the **top 5 most relevant clinical trials** based on semantic similarity, helping clinicians quickly assess suitable enrollment options.
+        """)
+        
+        gr.Markdown("## Workflow")
+        
+        # Row 1: Steps 1-3
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("""
+                <div style="background: linear-gradient(135deg, #fef3c7, #fde68a); border: 2px solid #f59e0b; border-radius: 12px; padding: 15px; text-align: center; margin: 5px;">
+                <div style="font-size: 2em;">📝</div>
+                <strong>1. Enter Patient Summary</strong><br>
+                <small>Describe the patient's history and findings.</small>
+                </div>
+                """)
+            with gr.Column():
+                gr.Markdown("""
+                <div style="background: linear-gradient(135deg, #ddd6fe, #c4b5fd); border: 2px solid #8b5cf6; border-radius: 12px; padding: 15px; text-align: center; margin: 5px;">
+                <div style="font-size: 2em;">🔑</div>
+                <strong>2. Keyword Extraction</strong><br>
+                <small>AI extracts key medical conditions using GPT-4-turbo.</small>
+                </div>
+                """)
+            with gr.Column():
+                gr.Markdown("""
+                <div style="background: linear-gradient(135deg, #bfdbfe, #93c5fd); border: 2px solid #3b82f6; border-radius: 12px; padding: 15px; text-align: center; margin: 5px;">
+                <div style="font-size: 2em;">🔎</div>
+                <strong>3. Trial Retrieval</strong><br>
+                <small>Keywords are matched to all trials (including Pfizer).</small>
+                </div>
+                """)
+        
+        # Row 2: Steps 4-5
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("""
+                <div style="background: linear-gradient(135deg, #bbf7d0, #86efac); border: 2px solid #10b981; border-radius: 12px; padding: 15px; text-align: center; margin: 5px;">
+                <div style="font-size: 2em;">✅</div>
+                <strong>4. Eligibility Check</strong><br>
+                <small>Inclusion/exclusion criteria are checked for each trial.</small>
+                </div>
+                """)
+            with gr.Column():
+                gr.Markdown("""
+                <div style="background: linear-gradient(135deg, #fecaca, #fca5a5); border: 2px solid #ef4444; border-radius: 12px; padding: 15px; text-align: center; margin: 5px;">
+                <div style="font-size: 2em;">🏆</div>
+                <strong>5. Ranking & Results</strong><br>
+                <small>Top 5 matching trials are ranked and displayed. Pfizer trials are highlighted if present.</small>
+                </div>
+                """)
+            with gr.Column():
+                gr.Markdown("")  # Empty column for better spacing
         
-        # Simple chat interface
+        gr.Markdown("## Start Matching")
         chatbot = gr.ChatInterface(
             fn=chat_function,
             examples=[
                 ["A 39-year-old Thai male presents for study screening. He received his second dose of an approved mRNA COVID-19 vaccine (Pfizer) over 4 months ago and has remained in good health since. He has no history of COVID-19 infection, confirmed by a negative SARS-CoV-2 rapid antigen test today. He denies any chronic illness, does not take any medications, and has never had an adverse reaction to vaccines. He is sexually active and reports consistent condom use with his partner. He is interested in participating in a study assessing immune response to a new booster dose."],
                 ["A 19-year-old pregnant patient, receiving prenatal care at Johns Hopkins Hospital, is 28 weeks gestation and preparing to receive her first mRNA COVID-19 vaccine dose. She has no previous history of COVID-19 and no significant medical history. She meets eligibility criteria and is interested in contributing to research on pregnancy and vaccine-related immune responses."],
                 ["A 67-year-old male with relapsed/refractory multiple myeloma presents for enrollment into a post-trial access study. He previously participated in a Pfizer-sponsored parent study evaluating elranatamab, during which he achieved a partial response and remained clinically stable. At the time the parent trial ended, he was continuing on elranatamab with no evidence of disease progression or significant toxicity. He reports no history of psychiatric illness or lab abnormalities, and he wishes to continue treatment through this access program."]
-            ],
-            title="Clinical Trial Matcher",
-            description="Enter patient details to find matching clinical trials"
+            ]
         )
     
     demo.launch(share=True)