Update app.py

app.py

@@ -116,8 +116,8 @@ write = Task(
 analyse = Task(
     description=(
         "1. Perform statistical analysis of patent filing trends, innovation hot spots, and growth projections.\n"
-        "2. Collaborate with the Planner and Writer to align on data needs.\n"
-        "3. Present findings in an actionable format."
+        "2. Provide structured output with fields 'Category' and 'Values' for visualization.\n"
+        "3. Collaborate with the Planner and Writer to align on data needs."
     ),
     expected_output="A detailed statistical analysis with actionable insights for stakeholders.",
     agent=analyst
@@ -154,16 +154,30 @@ def create_visualizations(analyst_output):
     """Create visualizations for advanced insights."""
     if enable_custom_visualization and analyst_output:
         try:
-            data = pd.DataFrame(analyst_output)
-            if 'Category' in data.columns and 'Values' in data.columns:
-                st.markdown("### Advanced Visualization")
-                fig = px.bar(data, x="Category", y="Values", title="Patent Trends by Category")
-                st.plotly_chart(fig)
+            # Validate and format analyst output
+            if isinstance(analyst_output, list) and all(isinstance(item, dict) for item in analyst_output):
+                data = pd.DataFrame(analyst_output)
+                if 'Category' in data.columns and 'Values' in data.columns:
+                    st.markdown("### Advanced Visualization")
+                    fig = px.bar(data, x="Category", y="Values", title="Patent Trends by Category")
+                    st.plotly_chart(fig)
+                else:
+                    st.warning("Data does not have the required columns 'Category' and 'Values'.")
             else:
-                st.warning("Data does not have the required columns for visualization.")
+                st.warning("Analyst output is not in the correct format for visualization.")
         except Exception as e:
             st.error(f"Failed to create visualizations: {e}")
 
+def display_table(analyst_output):
+    """Display tabular data for detailed insights."""
+    if analyst_output and isinstance(analyst_output, list):
+        try:
+            data = pd.DataFrame(analyst_output)
+            st.markdown("### Data Table")
+            st.dataframe(data)
+        except Exception as e:
+            st.error(f"Failed to display table: {e}")
+
 if st.button("Generate Patent Insights"):
     with st.spinner('Processing...'):
         try:
@@ -199,6 +213,9 @@ if st.button("Generate Patent Insights"):
                     if enable_advanced_analysis:
                         create_visualizations(analyst_output)
 
+                    # Display tabular data
+                    display_table(analyst_output)
+
             # Display Token Usage and Execution Time
             st.success(f"Analysis completed in {elapsed_time:.2f} seconds.")
             token_usage = getattr(results, "token_usage", None)
@@ -213,4 +230,4 @@ if st.button("Generate Patent Insights"):
                     st.download_button("Download Report", data=report_file, file_name="Patent_Report.pdf")
 
         except Exception as e:
-            st.error(f"An error occurred during execution: {e}")
+            st.error(f"An error occurred during execution: {e}")