Update app.py

app.py

@@ -385,63 +385,36 @@ def display_table(analyst_output):
 
     return table_data
 
-
-
 def parse_analyst_output(raw_output):
-    structured_data = []
-    current_category = None
-    current_values = []
-
-    # Split raw output by line
-    lines = raw_output.split('\n')
-
-    for line in lines:
-        line = line.strip()
-
-        # Detect the start of a new category
-        if line.startswith("Category:"):
-            # Save the previous category and its values
-            if current_category and current_values:
-                structured_data.append({
-                    "Category": current_category,
-                    "Values": current_values if len(current_values) > 1 else current_values[0]
-                })
-            # Start processing the new category
-            current_category = line.replace("Category:", "").strip()
-            current_values = []
-
-        # Skip 'Values:' header
-        elif line.startswith("Values:"):
-            continue
-
-        # Process the values under the current category
-        elif line and current_category:
-            try:
-                # Attempt to convert the line into Python data (dict/list)
-                parsed_value = ast.literal_eval(line)
-                current_values.append(parsed_value)
-            except (ValueError, SyntaxError):
-                # If parsing fails, treat it as plain text
-                current_values.append(line)
+    key_insights = []
+    data_insights = []
 
-    # Save the last processed category
-    if current_category and current_values:
-        structured_data.append({
-            "Category": current_category,
-            "Values": current_values if len(current_values) > 1 else current_values[0]
-        })
+    structured_data = ast.literal_eval(raw_output) if isinstance(raw_output, str) else raw_output
 
-    return structured_data
+    for item in structured_data:
+        if item.get("Type") == "Key Insight":
+            key_insights.append(item["Values"])
+        elif item.get("Type") == "Data Insight":
+            data_insights.append(item)
+        else:
+            data_insights.append(item)  
 
+    return key_insights, data_insights
 
 # Main Execution Block
 if st.button("Generate Patent Insights"):
     with st.spinner('Processing...'):
         try:
+            # Start the timer
             start_time = time.time()
+
+            # Kick off the crew with user inputs
             results = crew.kickoff(inputs={"topic": patent_area, "stakeholder": stakeholder})
+
+            # Calculate elapsed time
             elapsed_time = time.time() - start_time
 
+            # Extract Writer's Output
             writer_output = getattr(results.tasks_output[2], "raw", "No details available.")
             if writer_output:
                 st.markdown("### Final Report")
@@ -449,34 +422,55 @@ if st.button("Generate Patent Insights"):
             else:
                 st.warning("No final report available.")
 
+            # Expandable section for detailed insights
             with st.expander("Explore Detailed Insights"):
                 tab1, tab2 = st.tabs(["Planner's Insights", "Analyst's Analysis"])
 
+                # Planner's Insights
                 with tab1:
                     planner_output = getattr(results.tasks_output[0], "raw", "No details available.")
                     st.write(planner_output)
 
+                # Analyst's Analysis
                 with tab2:
                     analyst_output = getattr(results.tasks_output[1], "raw", "No details available.")
                     st.write(analyst_output)
-                    # Convert raw text to structured data
-                    if isinstance(analyst_output, str):
-                        analyst_output = parse_analyst_output(analyst_output)
+
+                    # Convert raw analyst output into structured key/data insights
+                    key_insights, data_insights = parse_analyst_output(analyst_output)
                     st.subheader("Structured Analyst Output")
-                    st.write(analyst_output)
-                    
+                    st.write(data_insights)
 
+                    # Create Visualizations if enabled
                     charts = []
                     if enable_advanced_analysis:
-                        charts = create_visualizations(analyst_output)
+                        charts = create_visualizations(data_insights)
 
-                    table_data = display_table(analyst_output)
+                    # Display Data Tables
+                    table_data = display_table(data_insights)
 
+            # Notify that analysis is complete
             st.success(f"Analysis completed in {elapsed_time:.2f} seconds.")
-            pdf_path = generate_pdf_report(writer_output, charts=charts, table_data=table_data, metadata={"Technology Area": patent_area, "Stakeholder": stakeholder})
+
+            # Generate the PDF report with Key Insights and Data Insights
+            pdf_path = generate_pdf_report(
+                result=writer_output,
+                charts=charts,
+                table_data=data_insights,
+                metadata={"Technology Area": patent_area, "Stakeholder": stakeholder},
+                key_insights=key_insights  # 🔑 Pass key insights to the PDF
+            )
+
+            # Download button for the generated PDF
             with open(pdf_path, "rb") as report_file:
-                st.download_button("Download Report", data=report_file, file_name="Patent_Strategy_Report.pdf")
+                st.download_button(
+                    label="Download Report",
+                    data=report_file,
+                    file_name="Patent_Strategy_Report.pdf",
+                    mime="application/pdf"
+                )
 
         except Exception as e:
+            # Handle errors
             logging.error(f"An error occurred during execution: {e}")
-            st.error(f"An error occurred during execution: {e}")
+            st.error(f"An error occurred during execution: {e}")