Update app.py

app.py

@@ -29,11 +29,56 @@ qa_pipeline = pipeline(
 st.markdown(
     """
     <style>
-        .main { background-color: #f4f4f4; }
-        .stTextInput, .stFileUploader { border: 2px solid #ff4b4b; border-radius: 10px; }
-        .stButton>button { background-color: #ff4b4b; color: white; border-radius: 10px; }
-        .stDownloadButton>button { background-color: #4CAF50; color: white; border-radius: 10px; }
-        h1, h2, h3, h4, h5, h6, p { color: #333333; }
+        /* Main background and text colors */
+        .main { 
+            background-color: #f4f4f4;
+        }
+
+        /* Text input boxes - light blue theme */
+        .stTextInput>div>div>input, 
+        .stTextArea>div>div>textarea {
+            background-color: #e6f2ff;
+            border: 1px solid #b3d1ff;
+            border-radius: 8px;
+            color: #003366;
+        }
+
+        /* File uploader - matching style */
+        .stFileUploader>div>div {
+            background-color: #e6f2ff;
+            border: 1px solid #b3d1ff;
+            border-radius: 8px;
+        }
+
+        /* Buttons - keeping your original style */
+        .stButton>button {
+            background-color: #ff4b4b;
+            color: white;
+            border-radius: 10px;
+            border: none;
+        }
+
+        .stDownloadButton>button {
+            background-color: #4CAF50;
+            color: white;
+            border-radius: 10px;
+            border: none;
+        }
+
+        /* Text colors */
+        h1, h2, h3, h4, h5, h6 {
+            color: #003366;  /* Dark blue for headers */
+        }
+
+        p {
+            color: #336699;  /* Medium blue for paragraphs */
+        }
+
+        /* Dataframe styling */
+        .dataframe {
+            background-color: #e6f2ff;
+            border: 1px solid #b3d1ff;
+        }
     </style>
     """,
     unsafe_allow_html=True
@@ -81,15 +126,110 @@ if uploaded_file:
             result = qa_pipeline(question=question, context=context)
         st.success(f"📝 Answer: {result['answer']}")
 
-    # ----------------- Word Cloud -----------------
-    st.subheader("☁ Word Cloud of News Text")
-    text = " ".join(df['content'].tolist())
-    wordcloud = WordCloud(width=800, height=400, background_color="white").generate(text)
-
-    fig, ax = plt.subplots()
-    ax.imshow(wordcloud, interpolation="bilinear")
-    ax.axis("off")
-    st.pyplot(fig)
+# ----------------- Visualization Section -----------------
+
+    st.subheader("📊 Data Visualizations")
+
+    # Create two main columns (60/40 split)
+    main_col1, main_col2 = st.columns([3, 2])
+
+    with main_col1:
+        # ----------------- Topic Distribution -----------------
+        st.markdown("*Topic Distribution*")
+
+        # Create sub-columns for the charts
+        chart_col1, chart_col2 = st.columns(2)
+
+        with chart_col1:
+            # Compact Pie Chart
+            fig1, ax1 = plt.subplots(figsize=(4, 4))
+            df['class'].value_counts().plot.pie(
+                autopct='%1.1f%%',
+                startangle=90,
+                ax=ax1,
+                colors=['#ff9999', '#66b3ff', '#99ff99', '#ffcc99', '#c2c2f0'],
+                wedgeprops={'linewidth': 0.5, 'edgecolor': 'white'}
+            )
+            ax1.set_ylabel('')
+            st.pyplot(fig1, use_container_width=True)
+
+        with chart_col2:
+            # Compact Bar Chart
+            fig2, ax2 = plt.subplots(figsize=(4, 4))
+            df['class'].value_counts().plot.bar(
+                color=['#ff9999', '#66b3ff', '#99ff99', '#ffcc99', '#c2c2f0'],
+                ax=ax2,
+                width=0.7
+            )
+            ax2.set_xlabel('')
+            ax2.set_ylabel('Count')
+            plt.xticks(rotation=45, ha='right')
+            st.pyplot(fig2, use_container_width=True)
+
+    with main_col2:
+        # ----------------- Compact Word Cloud -----------------
+        st.markdown("*Word Cloud*")
+        text = " ".join(df['content'].tolist())
+        wordcloud = WordCloud(
+            width=300,
+            height=200,
+            background_color="white",
+            collocations=False,
+            max_words=100
+        ).generate(text)
+
+        fig3, ax3 = plt.subplots(figsize=(4, 3))
+        ax3.imshow(wordcloud, interpolation="bilinear")
+        ax3.axis("off")
+        st.pyplot(fig3, use_container_width=True)
+
+# ----------------- Detailed Stats (below) -----------------
+    with st.expander("📈 Detailed Statistics", expanded=False):
+        stats_col1, stats_col2 = st.columns(2)
+
+        with stats_col1:
+            st.write("*Category Breakdown:*")
+            stats_df = df['class'].value_counts().reset_index()
+            stats_df.columns = ['Category', 'Count']
+            stats_df['Percentage'] = (stats_df['Count'] / stats_df['Count'].sum() * 100).round(1)
+            st.dataframe(stats_df, height=200)
+
+        with stats_col2:
+            if 'date' in df.columns:
+                try:
+                    st.write("*Monthly Trends*")
+                    df['date'] = pd.to_datetime(df['date'])
+                    trends = df.groupby([df['date'].dt.to_period('M'), 'class']).size().unstack()
+                    st.line_chart(trends)
+                except:
+                    st.warning("Date parsing failed")
+
+# ----------------- News Category Explorer -----------------
+    st.subheader("🔍 Explore News by Category")
+
+    # Get unique categories
+    categories = df['class'].unique()
+
+    # Create 5 columns for category buttons
+    cols = st.columns(5)
+
+    # Create a dictionary to store category articles
+    category_articles = {category: df[df['class'] == category] for category in categories}
+
+    # Place each category button in its own column
+    for i, category in enumerate(categories):
+        with cols[i]:
+            if st.button(category, key=f"btn_{category}"):
+                # Create pop-up window
+                with st.popover(f"📰 Articles in {category}", use_container_width=True):
+                    st.markdown(f"### {category} Articles")
+                    articles = category_articles[category]
+
+                    # Display articles with expandable content
+                    for idx, row in articles.iterrows():
+                        with st.expander(f"Article {idx + 1}: {row['content'][:50]}...", expanded=False):
+                            st.write(row['content'])
+                            st.caption(f"Classification confidence: {classifier(row['content'])[0]['score']:.2f}")
 
 # ----------------- Footer -----------------
 st.markdown("---")