Update app.py

app.py

@@ -138,22 +138,22 @@ st.markdown(
     .big-title {
         font-size: 36px !important;
         font-weight: bold;
-        color: #4A90E2;
+        color: var(--primary-color);
         text-align: center;
         margin-top: 20px;
         position: sticky; /* Pin the header */
         top: 0;
-        background-color: white; /* Ensure the header covers content when scrolling */
+        background-color: var(--background-color); /* Ensure the header covers content when scrolling */
         z-index: 100; /* Ensure it's above other content */
     }
     .sub-title {
         font-size: 20px;
-        color: #666;
+        color: var(--text-color);
         text-align: center;
         margin-bottom: 20px;
     }
     .stButton>button {
-        background-color: #4CAF50;
+        background-color: var(--primary-color);
         color: white;
         font-size: 16px;
         width: 100%;
@@ -163,13 +163,16 @@ st.markdown(
     .stTextArea textarea {
         font-size: 16px;
         min-height: 120px;
+        background-color: var(--secondary-background-color);
+        color: var(--text-color);
     }
     .result-box {
-        background-color: #f9f9f9;
+        background-color: var(--secondary-background-color);
         padding: 20px;
         border-radius: 10px;
         box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);
         margin-top: 20px;
+        color: var(--text-color);
     }
     .verdict {
         font-size: 24px;
@@ -177,6 +180,7 @@ st.markdown(
         margin: 0;
         display: flex;
         align-items: center;
+        color: var(--text-color);
     }
     .verdict-icon {
         margin-right: 10px;
@@ -185,12 +189,12 @@ st.markdown(
     div[data-baseweb="tab-list"] {
         position: sticky;
         top: 55px; /* Height of the header */
-        background-color: white;
+        background-color: var(--background-color);
         z-index: 99;
     }
     /* Fix the sidebar content */
     .stSidebar .sidebar-content {
-        background-color: #f0f2f6;
+        background-color: var(--secondary-background-color);
         padding: 20px;
         border-radius: 10px;
         position: sticky;
@@ -199,7 +203,7 @@ st.markdown(
         overflow-y: auto; /* Enable scrolling within the sidebar if content is too long */
     }
     .stSidebar .st-expander {
-        background-color: #ffffff;
+        background-color: var(--background-color);
         border-radius: 10px;
         padding: 10px;
         margin-bottom: 10px;
@@ -332,9 +336,9 @@ with st.container():
                     details = ""
                     if show_details:
                         details = f"""
-                            3-Class Probability: {result['prob3class'].item():.2f}
+                            3-Class Probability: {result['prob3class']:.2f}
                             3-Class Predicted Label: {['NEI', 'SUPPORTED', 'REFUTED'][result['pred_tc']]}
-                            2-Class Probability: {result['prob2class'].item():.2f}
+                            2-Class Probability: {result['prob2class']:.2f}
                             2-Class Predicted Label: {['SUPPORTED', 'REFUTED'][result['pred_bc']] if isinstance(result['pred_bc'], int) and result['pred_tc'] != 0 else 'Not used'}
                         """