Update thinktank/app.py

thinktank/app.py

@@ -7,6 +7,10 @@ import pandas as pd
 import plotly.express as px
 import tempfile
 import time
+import logging
+
+# Setup logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # Title and Application Introduction
 st.title("Patent Strategy and Innovation Consultant")
@@ -112,7 +116,7 @@ analyse = Task(
     description=(
         "1. Perform statistical analysis of patent filing trends, innovation hot spots, and growth projections.\n"
         "2. Identify top regions, key players, and technology combinations.\n"
-        "3. Generate visualizations such as heatmaps and multi-line charts for trends.\n"
+        "3. Generate visualizations such as heatmaps, bar charts, 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."
     ),
@@ -148,12 +152,17 @@ def generate_pdf_report(result, charts=None, table_data=None, metadata=None):
 
         if charts:
             for chart_path in charts:
-                pdf.add_page()
-                pdf.image(chart_path, x=10, y=20, w=180)
+                try:
+                    pdf.add_page()
+                    pdf.image(chart_path, x=10, y=20, w=180)
+                    logging.info(f"Successfully included chart: {chart_path}")
+                except Exception as e:
+                    logging.error(f"Failed to include chart in PDF: {chart_path}. Error: {e}")
 
         if table_data:
             pdf.add_page()
             pdf.set_font("Arial", size=10)
+            pdf.cell(200, 10, txt="Consolidated Table:", ln=True, align="L")
             for row in table_data:
                 pdf.cell(200, 10, txt=str(row), ln=True)
 
@@ -181,6 +190,9 @@ def create_visualizations(analyst_output):
     if validated_data:
         data = pd.DataFrame(validated_data)
         try:
+            if data.empty:
+                raise ValueError("Data for visualizations is empty.")
+
             bar_chart = px.bar(data, x="Category", y="Values", title="Patent Trends by Category")
             st.plotly_chart(bar_chart)
             with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_chart:
@@ -199,7 +211,14 @@ def create_visualizations(analyst_output):
                 heatmap_chart.write_image(temp_chart.name)
                 chart_paths.append(temp_chart.name)
 
+            multi_line_chart = px.line(data, x="Category", y="Values", title="Trends Over Time")
+            st.plotly_chart(multi_line_chart)
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_chart:
+                multi_line_chart.write_image(temp_chart.name)
+                chart_paths.append(temp_chart.name)
+
         except Exception as e:
+            logging.error(f"Error generating visualization: {e}")
             st.error(f"Error generating visualization: {e}")
     return chart_paths
 
@@ -250,4 +269,5 @@ if st.button("Generate Patent Insights"):
                 st.download_button("Download Report", data=report_file, file_name="Patent_Strategy_Report.pdf")
 
         except Exception as e:
-            st.error(f"An error occurred during execution: {e}")
+            logging.error(f"An error occurred during execution: {e}")
+            st.error(f"An error occurred during execution: {e}")