Update app.py

app.py

@@ -1,51 +1,3 @@
-import gradio as gr
-from transformers import pipeline
-import pandas as pd
-import os
-
-# Load the model
-classifier = pipeline(
-    "text-classification", 
-    model="ashishkgpian/biobert_icd9_classifier_ehr"
-)
-
-# Load ICD9 codes data
-icd9_data = pd.read_csv('D_ICD_DIAGNOSES.csv')
-icd9_data.columns = ['ROW_ID', 'ICD9_CODE', 'SHORT_TITLE', 'LONG_TITLE']
-
-def preprocessing(test_df):
-    test_df.loc[
-        test_df['ICD9_CODE'].str.startswith("V"), 'ICD9_CODE'] = test_df.ICD9_CODE.apply(
-        lambda x: x[:4])
-    test_df.loc[
-        test_df['ICD9_CODE'].str.startswith("E"), 'ICD9_CODE'] = test_df.ICD9_CODE.apply(
-        lambda x: x[:4])
-    test_df.loc[(~test_df.ICD9_CODE.str.startswith("E")) & (
-        ~test_df.ICD9_CODE.str.startswith("V")), 'ICD9_CODE'] = test_df.ICD9_CODE.apply(
-        lambda x: x[:3])
-    return test_df
-
-icd9_data = preprocessing(icd9_data)
-
-def classify_symptoms(text):
-    try:
-        results = classifier(text, top_k=5)
-        formatted_results = []
-        for result in results:
-            code = result['label']
-            code_info = icd9_data[icd9_data['ICD9_CODE'] == code]
-            formatted_results.append({
-                "ICD9 Code": code,
-                "Short Title": code_info['SHORT_TITLE'].iloc[0] if not code_info.empty else "N/A",
-                "Long Title": code_info['LONG_TITLE'].iloc[0] if not code_info.empty else "N/A",
-                "Confidence": f"{result['score']:.2%}"
-            })
-        return formatted_results
-    except Exception as e:
-        return f"Error processing classification: {str(e)}"
-
-# Enhanced CSS with white background and black text
-
 custom_css = """
 .gradio-container {
     width: 100% !important;
@@ -75,95 +27,89 @@ custom_css = """
     background: #000000;
     color: #ffffff;
 }
-h1 {
-    color: #b388ff !important;
-    font-size: 3rem !important;
-    margin-bottom: 0.5rem !important;
-    font-weight: 700 !important;
-}
-h3 {
-    color: #9575cd !important;
-    font-size: 1.4rem !important;
-    font-weight: 500 !important;
-    margin-bottom: 2rem !important;
-}
 .input-output-row {
     display: flex !important;
     gap: 2rem !important;
     margin: 2rem 0 !important;
 }
 .input-container {
-    background: #121212 !important;
-    padding: 2rem !important;
+    background: #1a1a1a !important;
+    padding: 1.5rem !important;
     border-radius: 12px !important;
-    box-shadow: 0 4px 6px rgba(255, 255, 255, 0.05) !important;
     width: 50% !important;
-    border: 1px solid #333333 !important;
     flex: 1 !important;
+    display: flex !important;
+    flex-direction: column !important;
+    gap: 1rem !important;
+}
+.inner-input-container {
+    background: #262626 !important;
+    padding: 1.5rem !important;
+    border-radius: 8px !important;
+    display: flex !important;
+    flex-direction: column !important;
+    gap: 1rem !important;
 }
 .input-container label {
     color: #ffffff !important;
-    font-weight: 600 !important;
-    font-size: 1.1rem !important;
+    font-weight: 500 !important;
+    font-size: 1rem !important;
     margin-bottom: 0.5rem !important;
     background: transparent !important;
 }
 textarea {
-    background: #1e1e1e !important;
+    background: #262626 !important;
     color: #ffffff !important;
-    border: 2px solid #673ab7 !important;
+    border: 2px solid #7c4dff !important;
     border-radius: 8px !important;
     padding: 1rem !important;
-    font-size: 1.2rem !important;
-    min-height: 150px !important;
+    font-size: 1rem !important;
+    min-height: 120px !important;
     width: 100% !important;
+    resize: none !important;
 }
 .submit-btn {
-    background-color: #673ab7 !important;
+    background-color: #7c4dff !important;
     color: white !important;
-    padding: 1rem 3rem !important;
+    padding: 0.75rem !important;
     border-radius: 8px !important;
-    font-size: 1.2rem !important;
-    margin-top: 1.5rem !important;
+    font-size: 1rem !important;
+    margin-top: 0.5rem !important;
     transition: all 0.3s ease !important;
-    width: auto !important;
-    font-weight: 600 !important;
+    width: 100% !important;
+    font-weight: 500 !important;
     border: none !important;
 }
 .submit-btn:hover {
-    background-color: #5e35b1 !important;
+    background-color: #6c3fff !important;
 }
 .output-container {
-    background: #121212 !important;
-    padding: 2rem !important;
+    background: #1a1a1a !important;
+    padding: 1.5rem !important;
     border-radius: 12px !important;
-    box-shadow: 0 4px 6px rgba(255, 255, 255, 0.05) !important;
     width: 50% !important;
-    border: 1px solid #333333 !important;
-    color: #ffffff !important;
     flex: 1 !important;
+    color: #ffffff !important;
 }
 .output-container label {
     color: #ffffff !important;
-    font-weight: 600 !important;
-    font-size: 1.1rem !important;
+    font-weight: 500 !important;
+    font-size: 1rem !important;
     margin-bottom: 1rem !important;
     background: transparent !important;
 }
 .examples-container {
-    background: #121212 !important;
-    padding: 2rem !important;
+    background: #1a1a1a !important;
+    padding: 1.5rem !important;
     border-radius: 12px !important;
     margin: 2rem 0 !important;
-    box-shadow: 0 4px 6px rgba(255, 255, 255, 0.05) !important;
     width: 100% !important;
-    border: 1px solid #333333 !important;
     color: #ffffff !important;
 }
 .examples-container label {
     color: #ffffff !important;
-    font-weight: 600 !important;
-    font-size: 1.1rem !important;
+    font-weight: 500 !important;
+    font-size: 1rem !important;
     background: transparent !important;
 }
 .footer {
@@ -180,20 +126,15 @@ textarea {
 with gr.Blocks(css=custom_css) as demo:
     with gr.Row(elem_classes=["main-container"]):
         with gr.Column(elem_classes=["content-wrapper"]):
-            gr.Markdown(
-                """
-                # 🏥 Clinical Symptom ICD9 Classifier
-                ### AI-Powered Medical Diagnosis Code Suggestion Tool
-                """
-            )
-    
             with gr.Row(elem_classes=["input-output-row"]):
                 with gr.Column(elem_classes=["input-container"]):
-                    input_text = gr.Textbox(
-                        label="Clinical Symptom Description",
-                        placeholder="Enter detailed patient symptoms and clinical observations...",
-                        lines=5
-                    )
+                    gr.Markdown("Clinical Symptom Description")
+                    with gr.Column(elem_classes=["inner-input-container"]):
+                        input_text = gr.Textbox(
+                            show_label=False,
+                            placeholder="Enter detailed patient symptoms and clinical observations...",
+                            lines=5
+                        )
                     submit_btn = gr.Button("Analyze Symptoms", elem_classes=["submit-btn"])
                 
                 with gr.Column(elem_classes=["output-container"]):