Update app.py

app.py

@@ -1,12 +1,11 @@
 ##############################################################################
 # Sozo Business Studio · 10-Jul-2025
-# • FIXED: Animation and FFmpeg errors while preserving the user's AI architecture.
-# • FIXED: The 'can't multiply sequence' error by replacing the animation engine.
-# • FIXED: FFmpeg failures with a robust media concatenation function.
+# • FIXED: Added missing 'inspect' import to resolve NameError in ChartGenerator.
+# • FIXED: Corrected tuple unpacking for pie charts in the animation function.
 # • NOTE: The user's prompts, classes, and AI calls are preserved exactly.
 ##############################################################################
 
-import os, re, json, hashlib, uuid, base64, io, tempfile, requests, subprocess
+import os, re, json, hashlib, uuid, base64, io, tempfile, requests, subprocess, inspect # FIXED: Added missing import
 from pathlib import Path
 from typing import Tuple, Dict, List
 
@@ -182,7 +181,6 @@ class ChartGenerator:
             if response.startswith("```json"): response = response[7:-3]
             elif response.startswith("```"): response = response[3:-3]
             spec_dict = json.loads(response)
-            # Filter to only include keys expected by the ChartSpecification constructor
             valid_keys = [p.name for p in inspect.signature(ChartSpecification).parameters.values()]
             filtered_dict = {k: v for k, v in spec_dict.items() if k in valid_keys}
             return ChartSpecification(**filtered_dict)
@@ -194,7 +192,7 @@ class ChartGenerator:
         numeric_cols = self.enhanced_ctx['numeric_columns']; categorical_cols = self.enhanced_ctx['categorical_columns']
         if "bar" in description.lower() and categorical_cols and numeric_cols: return ChartSpecification("bar", description, categorical_cols[0], numeric_cols[0])
         elif "pie" in description.lower() and categorical_cols and numeric_cols: return ChartSpecification("pie", description, categorical_cols[0], numeric_cols[0])
-        elif "line" in description.lower() and len(numeric_cols) >= 1: return ChartSpecification("line", description, self.df.columns[0], numeric_cols[0])
+        elif "line" in description.lower() and len(numeric_cols) >= 1: return ChartSpecification("line", self.df.columns[0], numeric_cols[0])
         elif "scatter" in description.lower() and len(numeric_cols) >= 2: return ChartSpecification("scatter", description, numeric_cols[0], numeric_cols[1])
         elif "hist" in description.lower() and numeric_cols: return ChartSpecification("hist", description, numeric_cols[0], None)
         else: return ChartSpecification("bar", description, self.df.columns[0], self.df.columns[1] if len(self.df.columns) > 1 else None)
@@ -229,14 +227,14 @@ def animate_chart(spec: ChartSpecification, df: pd.DataFrame, dur: float, out: P
     """FIXED: Renders a reliable animated chart using proven patterns, compatible with ChartSpecification."""
     plot_data = prepare_plot_data(spec, df)
     title = spec.title
-    frames = max(10, int(dur * fps)) # Ensure integer frame count
+    frames = max(10, int(dur * fps))
     fig, ax = plt.subplots(figsize=(WIDTH / 100, HEIGHT / 100), dpi=100)
     plt.tight_layout(pad=2.5)
     ctype = spec.chart_type
 
-    # This robust animation logic is adapted from the working example
     if ctype == "pie":
-        wedges, _ = ax.pie(plot_data, labels=plot_data.index, startangle=90, autopct='%1.1f%%')
+        # FIXED: Correctly unpack the three return values from ax.pie when autopct is used
+        wedges, _, _ = ax.pie(plot_data, labels=plot_data.index, startangle=90, autopct='%1.1f%%')
         ax.set_title(title); ax.axis('equal')
         def init(): [w.set_alpha(0) for w in wedges]; return wedges
         def update(i): [w.set_alpha(i / (frames - 1)) for w in wedges]; return wedges
@@ -249,7 +247,7 @@ def animate_chart(spec: ChartSpecification, df: pd.DataFrame, dur: float, out: P
             progress = i / (frames - 1)
             for b, h in zip(bars, plot_data.values): b.set_height(h * progress)
             return bars
-    else: # line, scatter, hist
+    else:
         line, = ax.plot([], [], lw=2)
         if ctype == 'scatter':
             x_full, y_full = plot_data.iloc[:, 0], plot_data.iloc[:, 1]
@@ -287,7 +285,6 @@ def safe_chart(desc: str, df: pd.DataFrame, dur: float, out: Path) -> str:
         return animate_chart(chart_spec, df, dur, out)
     except Exception as e:
         print(f"Chart animation failed for '{desc}': {e}. Falling back to static image.")
-        # Fallback: create a static version of the chart and fade it in
         temp_png = out.with_suffix(".png")
         llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", google_api_key=API_KEY, temperature=0.1)
         chart_generator = create_chart_generator(llm, df)
@@ -296,7 +293,7 @@ def safe_chart(desc: str, df: pd.DataFrame, dur: float, out: Path) -> str:
             img = cv2.imread(str(temp_png))
             img_resized = cv2.resize(img, (WIDTH, HEIGHT))
             return animate_image_fade(img_resized, dur, out)
-        else: # Ultimate fallback
+        else:
             img = generate_image_from_prompt(f"A professional business chart showing {desc}")
             img_cv = cv2.cvtColor(np.array(img.resize((WIDTH, HEIGHT))), cv2.COLOR_RGB2BGR)
             return animate_image_fade(img_cv, dur, out)