Update app.py

app.py

@@ -28,7 +28,7 @@ def preprocess_text(text):
 
 classifier_model = "Ginidu2003/Distilbert-Base-News-classifier"
 
-# ====================== COLORED BAR CHART ======================
+# ====================== BEAUTIFUL COLORED BAR CHART ======================
 def create_colored_bar_chart(category_counts):
     if category_counts is None or len(category_counts) == 0:
         fig, ax = plt.subplots()
@@ -38,17 +38,19 @@ def create_colored_bar_chart(category_counts):
     categories = category_counts["Category"]
     counts = category_counts["Count"]
 
-    colors = ['#FF6B6B', '#4ECDC4', '#45B7D1', '#96CEB4', '#FFEEAD']
+    # Nice modern color palette
+    colors = ['#3498DB', '#E67E22', '#9B59B6', '#2ECC71', '#E74C3C']
 
-    fig, ax = plt.subplots(figsize=(10, 6))
-    bars = ax.bar(categories, counts, color=colors)
+    fig, ax = plt.subplots(figsize=(11, 6))
+    bars = ax.bar(categories, counts, color=colors, edgecolor='white', linewidth=0.8)
 
+    # Add value on top of bars
     for bar in bars:
         height = bar.get_height()
-        ax.text(bar.get_x() + bar.get_width()/2, height + 0.5,
-                str(int(height)), ha='center', va='bottom', fontsize=12, fontweight='bold')
+        ax.text(bar.get_x() + bar.get_width()/2, height + 0.8,
+                str(int(height)), ha='center', va='bottom', fontsize=13, fontweight='bold')
 
-    ax.set_title("Category Distribution Across 5 Classes", fontsize=16, fontweight='bold')
+    ax.set_title("Category Distribution Across 5 Classes", fontsize=16, fontweight='bold', pad=20)
     ax.set_xlabel("Category", fontsize=12)
     ax.set_ylabel("Count", fontsize=12)
     plt.xticks(rotation=15)
@@ -116,14 +118,25 @@ def answer_question(news_content, question):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# ====================== BEAUTIFUL GRADIO INTERFACE ======================
+# ====================== BEAUTIFUL UI ======================
 with gr.Blocks(
     title="English News Classifier",
-    theme=gr.themes.Soft(),           # Beautiful modern theme
+    theme=gr.themes.Dark(),
     css="""
-    .gradio-container {max-width: 1100px; margin: auto;}
-    h1 {font-size: 2.5rem; text-align: center;}
-    .tab-label {font-size: 1.1rem; font-weight: 600;}
+    .gradio-container {
+        max-width: 1200px;
+        margin: auto;
+        background: linear-gradient(180deg, #1a1a2e, #16213e);
+    }
+    h1 {
+        text-align: center;
+        font-size: 2.8rem;
+        margin-bottom: 10px;
+    }
+    .tab-label {
+        font-size: 1.2rem;
+        font-weight: 600;
+    }
     """
 ) as demo:
 
@@ -131,16 +144,14 @@ with gr.Blocks(
     gr.Markdown("### Intelligent News Analysis Tool | Daily Mirror Sri Lanka")
 
     with gr.Tabs():
-        # ====================== CLASSIFICATION TAB ======================
         with gr.Tab("📊 News Classification"):
             gr.Markdown("### Upload CSV and get automatic category prediction")
             
-            with gr.Row():
-                file_input = gr.File(
-                    label="📤 Upload your CSV file",
-                    file_types=[".csv"],
-                    height=120
-                )
+            file_input = gr.File(
+                label="📤 Upload your CSV file",
+                file_types=[".csv"],
+                height=140
+            )
             
             classify_btn = gr.Button("🚀 Classify News", variant="primary", size="large")
             
@@ -156,20 +167,10 @@ with gr.Blocks(
                 outputs=[output_text, output_file, bar_chart]
             )
 
-        # ====================== Q&A TAB ======================
         with gr.Tab("❓ Question Answering"):
             gr.Markdown("### Ask any question about a news article")
-            
-            news_input = gr.Textbox(
-                lines=10,
-                label="📝 Paste News Content",
-                placeholder="Paste the full news article here..."
-            )
-            question_input = gr.Textbox(
-                label="❓ Your Question",
-                placeholder="e.g., What is the main issue? Who is involved?"
-            )
-            
+            news_input = gr.Textbox(lines=12, label="📝 Paste News Content", placeholder="Paste the full news article here...")
+            question_input = gr.Textbox(label="❓ Your Question", placeholder="e.g. What is the main topic?")
             qa_btn = gr.Button("🔍 Get Answer", variant="primary", size="large")
             qa_output = gr.Textbox(label="💡 Answer", lines=6)
 
@@ -180,6 +181,6 @@ with gr.Blocks(
             )
 
     gr.Markdown("---")
-    gr.Markdown("**Built for Text Analytics Assignment (In23-S5-DA3111) - Section 02**")
+    gr.Markdown("**Built for Text Analytics Assignment - Section 02**")
 
 demo.launch()