Update app.py

app.py

@@ -108,12 +108,10 @@ uploaded_file = "Intervention.csv"
 df_full = pd.read_csv(uploaded_file)  # Load the full data *once*
 st.write("Overall Data Visualizations")
 
+if tabs == 'Chat':
+    col1, col2 = st.columns([1, 1])  # Equal width columns
 
-
-if tabs =='Chat':
-    col1_full, col2_full = st.columns([0.9, 0.9])
-
-    with col1_full:
+    with col1:  # Visualizations on the left
         fig_gender_full = px.histogram(df_full, x="Gender", title="Overall Gender Distribution")
         fig_gender_full.update_layout(height=350, width=350)
         st.plotly_chart(fig_gender_full)
@@ -126,7 +124,6 @@ if tabs =='Chat':
         fig_company_full.update_layout(height=300, width=300)
         st.plotly_chart(fig_company_full)
 
-    with col2_full:
         fig_category_full = px.histogram(df_full, y="Intervention_Category", title="Overall Intervention Category Distribution")
         fig_category_full.update_layout(height=300, width=200)
         st.plotly_chart(fig_category_full)
@@ -138,23 +135,18 @@ if tabs =='Chat':
         fig_pie_category = px.pie(df_full, names='Intervention_Category', title='Overall Intervention Category Pie Chart')
         fig_pie_category.update_layout(height=400, width=500)
         st.plotly_chart(fig_pie_category)
-        
-    df = pd.read_csv(uploaded_file)
-    st.subheader("Chat with AI")
-    st.write("Get visualizations and analysis from our Gemini powered agent")
-        
-        # Read the CSV file
-    #df = pd.read_csv(uploaded_file)
-
-        # Display the data
-    with st.expander("Preview"):
-        st.write(df.head())
-
-        # Plot the data
-    user_input = st.text_input("Type your message here",placeholder="Ask me about your data")
-    if user_input:
-        answer = generateResponse(dataFrame=df,prompt=user_input)
-        st.write(answer)
+
+    with col2:  # Chat on the right
+        st.subheader("Chat with AI")
+        st.write("Get visualizations and analysis from our Gemini powered agent")
+
+        with st.expander("Preview"):
+            st.write(df.head())  # Assuming 'df' is defined earlier (e.g., from file upload)
+
+        user_input = st.text_input("Type your message here", placeholder="Ask me about your data")
+        if user_input:
+            answer = generateResponse(dataFrame=df, prompt=user_input)
+            st.write(answer)
         
 elif tabs == 'Reports':
     df = pd.read_csv(uploaded_file)
@@ -208,7 +200,7 @@ elif tabs == 'Reports':
                         response = model.generate_content(prompt)
                         report = response.text
                         st.markdown(report)
-                        st.success("Report Generation Complete (with potential errors)")
+                        st.success("Report Generation Complete")
                     except Exception as e:
                         st.write(f"Error generating report: {e}")