Update app.py

app.py

@@ -60,10 +60,16 @@ def classify_csv(file):
         
         output_file = "output.csv"
         df.to_csv(output_file, index=False)
-        
-        return f"✅ Success! Classified {len(df)} rows", output_file
+
+        # Count categories
+        category_counts = df['class'].value_counts().reset_index()
+        category_counts.columns = ["Category", "Count"]
+
+        return f"✅ Success! Classified {len(df)} rows", output_file, category_counts
     except Exception as e:
-        return f"❌ Error: {str(e)}", None
+        return f"❌ Error: {str(e)}", None, None
+        
+        
 
 # ====================== Q&A FUNCTION ======================
 def answer_question(news_content, question):
@@ -89,11 +95,17 @@ with gr.Blocks(title="Daily Mirror News Classifier") as demo:
             classify_btn = gr.Button("🚀 Classify News", variant="primary")
             output_text = gr.Textbox(label="Status")
             output_file = gr.File(label="Download output.csv")
+            bar_chart = gr.BarPlot(
+                 label="Category Distribution",
+                 x="Category",
+                 y="Count",
+                 vertical=True
+            )
             
             classify_btn.click(
                 fn=classify_csv,
                 inputs=file_input,
-                outputs=[output_text, output_file]
+                outputs=[output_text, output_file,bar_chart]
             )
 
         with gr.Tab("❓ Question Answering"):