Update app.py

app.py

@@ -65,7 +65,10 @@ def classify_csv(file):
         category_counts = df['class'].value_counts().reset_index()
         category_counts.columns = ["Category", "Count"]
 
-        return f"✅ Success! Classified {len(df)} rows", output_file, category_counts
+        # Create colored bar chart
+        fig = create_colored_bar_chart(category_counts)
+        
+        return f"✅ Success! Classified {len(df)} rows", output_file, fig
     except Exception as e:
         return f"❌ Error: {str(e)}", None, None
 
@@ -143,9 +146,8 @@ with gr.Blocks(title=" News Classifier & Question Answering App") as demo:
             output_text = gr.Textbox(label="Status")
             output_file = gr.File(label="Download output.csv")
             bar_chart = gr.Plot(
-                 label="Category Distribution Across 5 Classes",
-                 x="Category",
-                 y="Count",
+                 label="Category Distribution Across 5 Classes"
+                
                  
             )