Update app.py

app.py

@@ -87,32 +87,6 @@ if st.session_state.df is not None and st.session_state.show_preview:
     st.dataframe(st.session_state.df.head())
 
 
-#def ask_gpt4o_for_visualization(query, df, llm):
-#    columns = ', '.join(df.columns)
-#    prompt = f"""
-#    Analyze the query and suggest one or more relevant visualizations.
-#    Query: "{query}"
-#    Available Columns: {columns}
-#    Respond in this JSON format (as a list if multiple suggestions):
-#    [
-#      {{
-#       "chart_type": "bar/box/line/scatter",
-#        "x_axis": "column_name",
-#        "y_axis": "column_name",
-#        "group_by": "optional_column_name"
-#      }}
-#    ]
-#    """
-#    response = llm.generate(prompt)
-#    try:
-#        return json.loads(response)
-#    except json.JSONDecodeError:
-#        st.error("⚠️ GPT-4o failed to generate a valid suggestion.")
-#        return None
-
-
-
-
 # Helper Function for Validation
 def is_valid_suggestion(suggestion):
     chart_type = suggestion.get("chart_type", "").lower()
@@ -455,91 +429,6 @@ def handle_visualization_suggestions(suggestions, df):
         st.plotly_chart(fig, use_container_width=True)
 
 
-
-# Function to create TXT file
-def create_text_report_with_viz_temp(report, conclusion, visualizations):
-    content = f"### Analysis Report\n\n{report}\n\n### Visualizations\n"
-
-    for i, fig in enumerate(visualizations, start=1):
-        fig_title = fig.layout.title.text if fig.layout.title.text else f"Visualization {i}"
-        x_axis = fig.layout.xaxis.title.text if fig.layout.xaxis.title.text else "X-axis"
-        y_axis = fig.layout.yaxis.title.text if fig.layout.yaxis.title.text else "Y-axis"
-
-        content += f"\n{i}. {fig_title}\n"
-        content += f"   - X-axis: {x_axis}\n"
-        content += f"   - Y-axis: {y_axis}\n"
-
-        if fig.data:
-            trace_types = set(trace.type for trace in fig.data)
-            content += f"   - Chart Type(s): {', '.join(trace_types)}\n"
-        else:
-            content += "   - No data available in this visualization.\n"
-
-    content += f"\n\n\n{conclusion}"
-
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".txt", mode='w', encoding='utf-8') as temp_txt:
-        temp_txt.write(content)
-        return temp_txt.name
-
-
-
-# Function to create PDF with report text and visualizations
-def create_pdf_report_with_viz(report, conclusion, visualizations):
-    pdf = FPDF()
-    pdf.set_auto_page_break(auto=True, margin=15)
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-
-    # Title
-    pdf.set_font("Arial", style="B", size=18)
-    pdf.cell(0, 10, "📊 Analysis Report", ln=True, align="C")
-    pdf.ln(10)
-
-    # Report Content
-    pdf.set_font("Arial", style="B", size=14)
-    pdf.cell(0, 10, "Analysis", ln=True)
-    pdf.set_font("Arial", size=12)
-    pdf.multi_cell(0, 10, report)
-
-    pdf.ln(10)
-    pdf.set_font("Arial", style="B", size=14)
-    pdf.cell(0, 10, "Conclusion", ln=True)
-    pdf.set_font("Arial", size=12)
-    pdf.multi_cell(0, 10, conclusion)
-
-    # Add Visualizations
-    pdf.add_page()
-    pdf.set_font("Arial", style="B", size=16)
-    pdf.cell(0, 10, "📈 Visualizations", ln=True)
-    pdf.ln(5)
-
-    with tempfile.TemporaryDirectory() as temp_dir:
-        for i, fig in enumerate(visualizations, start=1):
-            fig_title = fig.layout.title.text if fig.layout.title.text else f"Visualization {i}"
-            x_axis = fig.layout.xaxis.title.text if fig.layout.xaxis.title.text else "X-axis"
-            y_axis = fig.layout.yaxis.title.text if fig.layout.yaxis.title.text else "Y-axis"
-
-            # Save each visualization as a PNG image
-            img_path = os.path.join(temp_dir, f"viz_{i}.png")
-            fig.write_image(img_path)
-
-            # Insert Title and Description
-            pdf.set_font("Arial", style="B", size=14)
-            pdf.multi_cell(0, 10, f"{i}. {fig_title}")
-            pdf.set_font("Arial", size=12)
-            pdf.multi_cell(0, 10, f"X-axis: {x_axis} | Y-axis: {y_axis}")
-            pdf.ln(3)
-
-            # Embed Visualization
-            pdf.image(img_path, w=170)
-            pdf.ln(10)
-
-        # Save PDF
-        temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
-        pdf.output(temp_pdf.name)
-
-    return temp_pdf
-
 def escape_markdown(text):
     # Ensure text is a string
     text = str(text)
@@ -684,28 +573,6 @@ if st.session_state.df is not None:
                 safe_conclusion = escape_markdown(conclusion_result if conclusion_result else "⚠️ No Conclusion Generated.")
                 st.markdown(safe_conclusion)
 
-    # Full Data Visualization Tab
-    with tab2:
-        st.subheader("📊 Comprehensive Data Visualizations")
-
-        fig1 = px.histogram(st.session_state.df, x="job_title", title="Job Title Frequency")
-        st.plotly_chart(fig1)
-
-        fig2 = px.bar(
-            st.session_state.df.groupby("experience_level")["salary_in_usd"].mean().reset_index(),
-            x="experience_level", y="salary_in_usd",
-            title="Average Salary by Experience Level"
-        )
-        st.plotly_chart(fig2)
-
-        fig3 = px.box(st.session_state.df, x="employment_type", y="salary_in_usd",
-                      title="Salary Distribution by Employment Type")
-        st.plotly_chart(fig3)
-
-    temp_dir.cleanup()
-else:
-    st.info("Please load a dataset to proceed.")
-
 
 # Sidebar Reference
 with st.sidebar: