Update app.py

app.py

@@ -276,12 +276,43 @@ def create_visualizations(analyst_output):
 def display_table(analyst_output):
     table_data = []
     validated_data = validate_analyst_output(analyst_output)
+
     if validated_data:
-        data = pd.DataFrame(validated_data)
-        st.dataframe(data)
-        table_data = data.to_dict(orient="records")
+        for item in validated_data:
+            category = item["Category"]
+            values = item["Values"]
+
+            try:
+                # Dictionary → Structured Table
+                if isinstance(values, dict):
+                    df = pd.DataFrame(list(values.items()), columns=["Label", "Count"])
+                    st.subheader(f"{category} (Table View)")
+                    st.dataframe(df)
+                    table_data.extend(df.to_dict(orient="records"))
+
+                # List → Simple List Table
+                elif isinstance(values, list):
+                    df = pd.DataFrame(values, columns=["Items"])
+                    st.subheader(f"{category} (List View)")
+                    st.dataframe(df)
+                    table_data.extend(df.to_dict(orient="records"))
+
+                # String → Text Table
+                elif isinstance(values, str):
+                    st.subheader(f"{category} (Summary)")
+                    st.table(pd.DataFrame({"Insights": [values]}))
+                    table_data.append({"Category": category, "Values": values})
+
+                else:
+                    st.warning(f"Unsupported data format for category: {category}")
+
+            except Exception as e:
+                logging.error(f"Error displaying {category} as table: {e}")
+                st.error(f"Failed to display {category} as a table.")
+
     return table_data
 
+
 def parse_analyst_output(raw_output):
     structured_data = []
     current_category = None