Update app.py

app.py

@@ -8,116 +8,87 @@ classifier = pipeline(
 )
 
 def classify_symptoms(text):
-    """
-    Classify medical symptoms and return top ICD9 codes
-    
-    Args:
-        text (str): Input medical symptom description
-    
-    Returns:
-        dict: Top classification results with ICD9 codes and probabilities
-    """
     try:
-        # Get classification results
         results = classifier(text, top_k=5)
-        
-        # Format results for more readable output
         formatted_results = []
         for result in results:
             formatted_results.append({
                 "ICD9 Code": result['label'], 
                 "Confidence": f"{result['score']:.2%}"
             })
-        
         return formatted_results
-    
     except Exception as e:
         return f"Error processing classification: {str(e)}"
 
-# Custom CSS for medical-themed UI
+# Custom CSS without theme dependencies
 custom_css = """
-/* Medical-inspired color palette and design */
 .gradio-container {
-    background-color: #f4f7f6;
-    font-family: 'Arial', 'Helvetica Neue', sans-serif;
+    background-color: #f8f9fa;
 }
 
-/* Styled input area */
-.input-container {
-    background-color: #ffffff;
-    border: 2px solid #3498db;
-    border-radius: 10px;
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-    padding: 15px;
+.contain {
+    background-color: white !important;
+    border-radius: 8px !important;
+    padding: 20px !important;
+    box-shadow: 0 2px 6px rgba(0, 0, 0, 0.1) !important;
+    margin-bottom: 20px !important;
 }
 
-/* Result card styling */
-.result-card {
-    background-color: #ffffff;
-    border-left: 5px solid #2ecc71;
-    margin-bottom: 10px;
-    padding: 10px;
-    border-radius: 5px;
-    box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
+.textbox {
+    border: 2px solid #4a90e2 !important;
+    border-radius: 6px !important;
 }
 
-/* Medical icon-like styling for headers */
-.gradio-header {
-    color: #2c3e50;
-    text-align: center;
-    font-weight: bold;
+.label {
+    color: #2c3e50 !important;
+    font-weight: bold !important;
+    font-size: 1.1em !important;
 }
 
-/* Soften the medical green theme */
-.submit-button {
-    background-color: #2ecc71 !important;
-    color: white !important;
-    border: none !important;
-    transition: all 0.3s ease;
-}
-
-.submit-button:hover {
-    background-color: #27ae60 !important;
-    transform: scale(1.05);
+footer {
+    text-align: center;
+    padding: 20px;
+    color: #666;
+    font-size: 0.9em;
 }
 """
 
-# Create Gradio interface with advanced medical-themed design
-demo = gr.Interface(
-    fn=classify_symptoms,
-    inputs=gr.Textbox(
-        label="Patient Symptom Description", 
-        placeholder="Enter detailed patient symptoms (e.g., 'Patient reports persistent chest pain radiating to left arm, accompanied by shortness of breath')",
-        lines=4,
-        container=False,
-        elem_classes=["input-container"]
-    ),
-    outputs=gr.JSON(
-        label="Diagnostic Suggestions", 
-        elem_classes=["result-card"]
-    ),
-    title="Symptom-to-ICD9 Classifier",
-    description="An advanced AI-powered diagnostic assistance tool for converting clinical symptom descriptions into potential ICD9 diagnostic codes.",
-    theme="huggingface",
-    css=custom_css,
-    examples=[
-        ["45-year-old male experiencing severe chest pain, radiating to left arm, with shortness of breath and excessive sweating"],
-        ["Persistent headache for 2 weeks, accompanied by dizziness and occasional blurred vision"],
-        ["Diabetic patient reporting frequent urination, increased thirst, and unexplained weight loss"],
-        ["Elderly patient with chronic knee pain, reduced mobility, and signs of inflammation"]
-    ]
-)
-
-# Add some footer-like information
-demo.layout = gr.Column(
-    [
-        demo.title,
-        demo.description,
-        demo.input,
-        demo.output,
-        gr.Markdown("**Disclaimer:** This is an AI-assisted diagnostic tool. Always consult with a healthcare professional for accurate diagnosis and treatment.")
-    ]
-)
+with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown(
+        """
+        # MedAI: Clinical Symptom ICD9 Classifier
+        ### AI-Powered Diagnostic Code Suggestion Tool
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column():
+            input_text = gr.Textbox(
+                label="Patient Symptom Description",
+                placeholder="Enter detailed patient symptoms (e.g., 'Patient reports persistent chest pain radiating to left arm, accompanied by shortness of breath')",
+                lines=4
+            )
+            
+            examples = gr.Examples(
+                examples=[
+                    ["45-year-old male experiencing severe chest pain, radiating to left arm, with shortness of breath and excessive sweating"],
+                    ["Persistent headache for 2 weeks, accompanied by dizziness and occasional blurred vision"],
+                    ["Diabetic patient reporting frequent urination, increased thirst, and unexplained weight loss"],
+                    ["Elderly patient with chronic knee pain, reduced mobility, and signs of inflammation"]
+                ],
+                inputs=input_text
+            )
+    
+    output = gr.JSON(label="Diagnostic Suggestions")
+    
+    input_text.submit(fn=classify_symptoms, inputs=input_text, outputs=output)
+    
+    gr.Markdown(
+        """
+        ---
+        **Disclaimer:** This is an AI-assisted diagnostic tool. Always consult with a healthcare professional for accurate diagnosis and treatment.
+        """
+    )
 
 if __name__ == "__main__":
     demo.launch()