Update app.py

app.py

@@ -151,32 +151,6 @@ def generate_visualization(suggestion, df):
         st.error(f"⚠️ Failed to generate visualization: {e}")
         return None
 
-# 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
-
-
 def add_stats_to_figure(fig, df, y_axis, chart_type):
     # Calculate statistics
     min_val = df[y_axis].min()
@@ -230,6 +204,33 @@ def add_stats_to_figure(fig, df, y_axis, chart_type):
     return fig
 
 
+# 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()