Update app.py

app.py

@@ -90,9 +90,10 @@ plan = Task(
         "1. Research recent trends in {topic} patent filings and innovation.\n"
         "2. Identify key players and emerging technologies.\n"
         "3. Provide recommendations for stakeholders on strategic directions.\n"
-        "4. Limit the output to 500 words."
+        "4. Identify key statistics such as top regions, top players, and hot areas of innovation.\n"
+        "5. Limit the output to 500 words."
     ),
-    expected_output="A research document with structured insights and strategic recommendations.",
+    expected_output="A research document with structured insights, strategic recommendations, and key statistics.",
     agent=planner
 )
 
@@ -100,20 +101,23 @@ write = Task(
     description=(
         "1. Use the Planner's and Analyst's outputs to craft a professional patent insights document.\n"
         "2. Include key findings, visual aids, and actionable strategies.\n"
-        "3. Align content with stakeholder goals.\n"
-        "4. Limit the document to 400 words."
+        "3. Suggest strategic directions and highlight untapped innovation areas.\n"
+        "4. Incorporate summarized tables for key statistics and example inventions.\n"
+        "5. Limit the document to 400 words."
     ),
-    expected_output="A polished, stakeholder-ready patent insights document.",
+    expected_output="A polished, stakeholder-ready patent insights document with actionable recommendations.",
     agent=writer
 )
 
 analyse = Task(
     description=(
         "1. Perform statistical analysis of patent filing trends, innovation hot spots, and growth projections.\n"
-        "2. Provide structured output with fields 'Category' and 'Values' for visualization.\n"
-        "3. Collaborate with the Planner and Writer to align on data needs."
+        "2. Identify top regions, key players, and technology combinations.\n"
+        "3. Generate visualizations such as heatmaps and multi-line charts for trends.\n"
+        "4. Provide structured output with fields 'Category' and 'Values' for visualization.\n"
+        "5. Collaborate with the Planner and Writer to align on data needs."
     ),
-    expected_output="A detailed statistical analysis with actionable insights for stakeholders.",
+    expected_output="A detailed statistical analysis with actionable insights, heatmaps, and trends.",
     agent=analyst
 )
 
@@ -190,6 +194,12 @@ def create_visualizations(analyst_output):
                 pie_chart.write_image(temp_chart.name)
                 chart_paths.append(temp_chart.name)
 
+            heatmap_chart = px.density_heatmap(data, x="Category", y="Values", title="Regional Patent Density")
+            st.plotly_chart(heatmap_chart)
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_chart:
+                heatmap_chart.write_image(temp_chart.name)
+                chart_paths.append(temp_chart.name)
+
         except Exception as e:
             st.error(f"Error generating visualization: {e}")
     return chart_paths