Update app.py

app.py

@@ -12,20 +12,22 @@ classifier = pipeline(
 # Load ICD9 codes data
 icd9_data = pd.read_csv('D_ICD_DIAGNOSES.csv')
 
+icd9_data.columns = ['ROW_ID', 'ICD9_CODE', 'SHORT_TITLE', 'LONG_TITLE']
 
-icd9_data.columns = ['ROW_ID', 'short_codes', 'SHORT_TITLE', 'LONG_TITLE']
 def preprocessing(test_df) : 
     test_df.loc[
-        test_df['short_codes'].str.startswith("V"), 'short_codes'] = test_df.short_codes.apply(
+        test_df['ICD9_CODE'].str.startswith("V"), 'ICD9_CODE'] = test_df.ICD9_CODE.apply(
         lambda x: x[:4])
     test_df.loc[
-        test_df['short_codes'].str.startswith("E"), 'short_codes'] = test_df.short_codes.apply(
+        test_df['ICD9_CODE'].str.startswith("E"), 'ICD9_CODE'] = test_df.ICD9_CODE.apply(
         lambda x: x[:4])
-    test_df.loc[(~test_df.short_codes.str.startswith("E")) & (
-        ~test_df.short_codes.str.startswith("V")), 'short_codes'] = test_df.short_codes.apply(
+    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):
@@ -46,21 +48,44 @@ def classify_symptoms(text):
     except Exception as e:
         return f"Error processing classification: {str(e)}"
 
-# Enhanced CSS with violet theme and better text contrast
+# Enhanced CSS with full-screen layout and violet theme
 custom_css = """
+html, body {
+    margin: 0 !important;
+    padding: 0 !important;
+    height: 100% !important;
+    overflow: hidden !important;
+}
 .gradio-container {
-    max-width: 1200px !important;
-    margin: auto !important;
-    padding: 2rem !important;
+    width: 100vw !important;
+    height: 100vh !important;
+    max-width: none !important;
+    margin: 0 !important;
+    padding: 0 !important;
+    display: flex !important;
+    flex-direction: column !important;
     background-color: #f5f3f7 !important;
 }
 .main-container {
     text-align: center;
     padding: 1rem;
-    margin-bottom: 2rem;
     background: #ffffff;
-    border-radius: 10px;
-    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
+}
+#component-1 {
+    height: 100% !important;
+    display: flex !important;
+    flex-direction: column !important;
+}
+.block {
+    flex-grow: 1 !important;
+    display: flex !important;
+    flex-direction: column !important;
+}
+.input-container, .output-container {
+    flex-grow: 1 !important;
+    display: flex !important;
+    flex-direction: column !important;
+    justify-content: center !important;
 }
 h1 {
     color: #4a148c !important;
@@ -75,18 +100,18 @@ h3 {
 .input-container {
     background: white !important;
     padding: 2rem !important;
-    border-radius: 10px !important;
-    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1) !important;
-    margin-bottom: 1.5rem !important;
+    margin: 0 !important;
 }
 textarea {
+    width: 100% !important;
     background: white !important;
     color: #000000 !important;
     border: 2px solid #7b1fa2 !important;
     border-radius: 8px !important;
     padding: 1rem !important;
     font-size: 1.1rem !important;
-    min-height: 120px !important;
+    min-height: 200px !important;
+    flex-grow: 1 !important;
 }
 .submit-btn {
     background-color: #6a1b9a !important;
@@ -103,32 +128,26 @@ textarea {
 .output-container {
     background: white !important;
     padding: 2rem !important;
-    border-radius: 10px !important;
-    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1) !important;
+    flex-grow: 1 !important;
+    overflow-y: auto !important;
 }
 .output-container pre {
     background: #f8f9fa !important;
     color: #000000 !important;
     border-radius: 8px !important;
     padding: 1rem !important;
+    max-height: 100% !important;
+    overflow-y: auto !important;
 }
 .examples-container {
     background: white !important;
     padding: 1.5rem !important;
-    border-radius: 10px !important;
     margin-top: 1rem !important;
-    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1) !important;
-}
-.example-text {
-    color: #000000 !important;
 }
 .footer {
     text-align: center;
-    margin-top: 2rem;
     padding: 1rem;
     background: white;
-    border-radius: 10px;
-    box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);
     color: #4a148c;
 }
 """
@@ -138,12 +157,11 @@ with gr.Blocks(css=custom_css) as demo:
         gr.Markdown(
             """
             # 🏥 Clinical Symptom ICD9 Classifier
-            
             """
         )
     
-    with gr.Row():
-        with gr.Column(elem_classes=["input-container"]):
+    with gr.Row(style="flex-grow: 1;"):
+        with gr.Column(elem_classes=["input-container"], scale=1):
             input_text = gr.Textbox(
                 label="Clinical Symptom Description",
                 placeholder="Enter detailed patient symptoms and clinical observations...",
@@ -151,7 +169,7 @@ with gr.Blocks(css=custom_css) as demo:
             )
             submit_btn = gr.Button("Analyze Symptoms", elem_classes=["submit-btn"])
             
-    with gr.Row(elem_classes=["output-container"]):
+    with gr.Row(elem_classes=["output-container"], style="flex-grow: 1;"):
         output = gr.JSON(
             label="Suggested ICD9 Diagnostic Codes with Descriptions"
         )