Update app.py

app.py

@@ -5,6 +5,8 @@ from langchain_groq import ChatGroq
 from fpdf import FPDF
 import pandas as pd
 import plotly.express as px
+import matplotlib.pyplot as plt
+import tempfile
 import time
 
 # Title and Sidebar
@@ -129,29 +131,43 @@ crew = Crew(
     verbose=True
 )
 
-def generate_pdf_report(result):
+def generate_pdf_report(result, charts=None, table_data=None):
     """Generate a professional PDF report from the Crew output."""
-    pdf = FPDF()
-    pdf.add_page()
-    pdf.set_font("Arial", size=12)
-    pdf.set_auto_page_break(auto=True, margin=15)
-
-    # Title
-    pdf.set_font("Arial", size=16, style="B")
-    pdf.cell(200, 10, txt="Patent Insights Report", ln=True, align="C")
-    pdf.ln(10)
-
-    # Content
-    pdf.set_font("Arial", size=12)
-    pdf.multi_cell(0, 10, txt=result)
-
-    # Save PDF
-    report_path = "Patent_Insights_Report.pdf"
-    pdf.output(report_path)
-    return report_path
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
+        pdf = FPDF()
+        pdf.add_page()
+        pdf.set_font("Arial", size=12)
+        pdf.set_auto_page_break(auto=True, margin=15)
+
+        # Title
+        pdf.set_font("Arial", size=16, style="B")
+        pdf.cell(200, 10, txt="Patent Insights Report", ln=True, align="C")
+        pdf.ln(10)
+
+        # Content
+        pdf.set_font("Arial", size=12)
+        pdf.multi_cell(0, 10, txt=result)
+
+        # Add charts if provided
+        if charts:
+            for chart_path in charts:
+                pdf.add_page()
+                pdf.image(chart_path, x=10, y=20, w=180)
+
+        # Add tables if provided
+        if table_data:
+            pdf.add_page()
+            pdf.set_font("Arial", size=10)
+            for row in table_data:
+                pdf.cell(200, 10, txt=str(row), ln=True)
+
+        # Save PDF
+        pdf.output(temp_pdf.name)
+        return temp_pdf.name
 
 def create_visualizations(analyst_output):
     """Create visualizations for advanced insights."""
+    chart_paths = []
     if enable_custom_visualization and analyst_output:
         try:
             st.markdown("#### Debugging Analyst Output")
@@ -162,17 +178,26 @@ def create_visualizations(analyst_output):
                 data = pd.DataFrame(analyst_output)
                 if 'Category' in data.columns and 'Values' in data.columns:
                     st.markdown("### Advanced Visualization")
+
+                    # Create a bar chart
                     fig = px.bar(data, x="Category", y="Values", title="Patent Trends by Category")
                     st.plotly_chart(fig)
+
+                    # Save chart to a file for embedding in PDF
+                    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_chart:
+                        fig.write_image(temp_chart.name)
+                        chart_paths.append(temp_chart.name)
                 else:
                     st.warning("Data does not have the required columns 'Category' and 'Values'.")
             else:
                 st.warning("Analyst output is not in the correct format for visualization.")
         except Exception as e:
             st.error(f"Failed to create visualizations: {e}")
+    return chart_paths
 
 def display_table(analyst_output):
     """Display tabular data for detailed insights."""
+    table_data = []
     if analyst_output and isinstance(analyst_output, list):
         try:
             st.markdown("#### Debugging Analyst Output for Table")
@@ -181,8 +206,10 @@ def display_table(analyst_output):
             data = pd.DataFrame(analyst_output)
             st.markdown("### Data Table")
             st.dataframe(data)
+            table_data = data.to_dict(orient="records")
         except Exception as e:
             st.error(f"Failed to display table: {e}")
+    return table_data
 
 if st.button("Generate Patent Insights"):
     with st.spinner('Processing...'):
@@ -216,11 +243,12 @@ if st.button("Generate Patent Insights"):
                     st.write(analyst_output)
 
                     # Generate visualizations if enabled
+                    charts = []
                     if enable_advanced_analysis:
-                        create_visualizations(analyst_output)
+                        charts = create_visualizations(analyst_output)
 
                     # Display tabular data
-                    display_table(analyst_output)
+                    table_data = display_table(analyst_output)
 
             # Display Token Usage and Execution Time
             st.success(f"Analysis completed in {elapsed_time:.2f} seconds.")
@@ -231,9 +259,9 @@ if st.button("Generate Patent Insights"):
 
             # Generate PDF Report
             if writer_output:
-                report_path = generate_pdf_report(writer_output)
-                with open(report_path, "rb") as report_file:
+                pdf_path = generate_pdf_report(writer_output, charts=charts, table_data=table_data)
+                with open(pdf_path, "rb") as report_file:
                     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}")