Update test.py

test.py

@@ -2,22 +2,53 @@ import streamlit as st
 from crewai import Agent, Task, Crew
 import os
 from langchain_groq import ChatGroq
+from langchain_openai import ChatOpenAI
 from fpdf import FPDF
 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 Insights Consultant")
+st.title("Patent Strategy and Innovation Consultant")
 st.sidebar.write(
-    "This Patent Insights Consultant uses a multi-agent system to provide actionable insights and data analysis for the patent domain."
+    "This application uses AI to provide actionable insights and comprehensive analysis for patent-related strategies."
 )
 
 # User Input Section
 st.sidebar.header("User Inputs")
-patent_area = st.text_input('Enter Patent Technology Area', value="Transparent Antennas for Windshields")
-stakeholder = st.text_input('Enter Stakeholder', value="Patent Attorneys")
+patent_area = st.text_input("Enter Patent Technology Area", value="Transparent Antennas for Windshields")
+stakeholder = st.text_input("Enter Stakeholder", value="Patent Attorneys")
+
+# Initialize LLM
+llm = None
+
+# Model Selection
+#st.header("Model Selection")
+model_choice = st.selectbox("Select LLM", ["OpenAI Model","Groq-based LLM"])
+
+# API Key Validation and LLM Initialization
+groq_api_key = os.getenv("GROQ_API_KEY")
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+#llm = ChatGroq(groq_api_key=os.getenv("GROQ_API_KEY"), model="groq/llama-3.3-70b-versatile")
+
+if model_choice == "Groq-based LLM":
+    if not groq_api_key:
+        st.error("Groq API key is missing. Please set the GROQ_API_KEY environment variable.")
+        llm = None
+    else:
+        llm = ChatGroq(groq_api_key=groq_api_key, model="groq/llama-3.3-70b-versatile")
+elif model_choice == "OpenAI Model":
+    if not openai_api_key:
+        st.error("OpenAI API key is missing. Please set the OPENAI_API_KEY environment variable.")
+        llm = None
+    else:
+        llm = ChatOpenAI(api_key=openai_api_key, model="gpt-4o")
 
 # Advanced Options
 st.sidebar.header("Advanced Options")
@@ -46,8 +77,6 @@ with st.sidebar.expander("Customize Agent Goals", expanded=False):
         writer_goal = "Craft a professional insights document summarizing trends, strategies, and actionable outcomes for stakeholders."
         analyst_goal = "Perform detailed statistical analysis of patent filings, growth trends, and innovation distribution."
 
-# LLM Initialization
-llm = ChatGroq(groq_api_key=os.getenv("GROQ_API_KEY"), model="groq/llama-3.3-70b-versatile")
 
 # Agent Definitions
 planner = Agent(
@@ -89,9 +118,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
 )
 
@@ -99,20 +129,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 650 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, 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."
     ),
-    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
 )
 
@@ -123,7 +156,6 @@ crew = Crew(
 )
 
 # PDF Report Generation
-
 def generate_pdf_report(result, charts=None, table_data=None, metadata=None):
     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
         pdf = FPDF()
@@ -132,7 +164,7 @@ def generate_pdf_report(result, charts=None, table_data=None, metadata=None):
         pdf.set_auto_page_break(auto=True, margin=15)
 
         pdf.set_font("Arial", size=16, style="B")
-        pdf.cell(200, 10, txt="Patent Insights Report", ln=True, align="C")
+        pdf.cell(200, 10, txt="Patent Strategy and Innovation Report", ln=True, align="C")
         pdf.ln(10)
 
         if metadata:
@@ -145,12 +177,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)
 
@@ -158,7 +195,6 @@ def generate_pdf_report(result, charts=None, table_data=None, metadata=None):
         return temp_pdf.name
 
 # Data Validation
-
 def validate_analyst_output(analyst_output):
     if not analyst_output:
         st.warning("No data available for analysis.")
@@ -173,19 +209,41 @@ def validate_analyst_output(analyst_output):
     return analyst_output
 
 # Visualization and Table Display
-
 def create_visualizations(analyst_output):
     chart_paths = []
     validated_data = validate_analyst_output(analyst_output)
     if validated_data:
         data = pd.DataFrame(validated_data)
         try:
-            fig = px.bar(data, x="Category", y="Values", title="Patent Trends by Category")
-            st.plotly_chart(fig)
+            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:
+                bar_chart.write_image(temp_chart.name)
+                chart_paths.append(temp_chart.name)
+
+            pie_chart = px.pie(data, names="Category", values="Values", title="Category Distribution")
+            st.plotly_chart(pie_chart)
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_chart:
+                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:
-                fig.write_image(temp_chart.name)
+                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
 
@@ -199,42 +257,50 @@ def display_table(analyst_output):
     return table_data
 
 # Main Execution Block
-
-if st.button("Generate Patent Insights"):
-    with st.spinner('Processing...'):
-        try:
-            start_time = time.time()
-            results = crew.kickoff(inputs={"topic": patent_area, "stakeholder": stakeholder})
-            elapsed_time = time.time() - start_time
-
-            writer_output = getattr(results.tasks_output[2], "raw", "No details available.")
-            if writer_output:
-                st.markdown("### Final Report")
-                st.write(writer_output)
-            else:
-                st.warning("No final report available.")
-
-            with st.expander("Explore Detailed Insights"):
-                tab1, tab2 = st.tabs(["Planner's Insights", "Analyst's Analysis"])
-
-                with tab1:
-                    planner_output = getattr(results.tasks_output[0], "raw", "No details available.")
-                    st.write(planner_output)
-
-                with tab2:
-                    analyst_output = getattr(results.tasks_output[1], "raw", "No details available.")
-                    st.write(analyst_output)
-
-                    charts = []
-                    if enable_advanced_analysis:
-                        charts = create_visualizations(analyst_output)
-
-                    table_data = display_table(analyst_output)
-
-            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})
-            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}")
+if st.button("Generate Insights"):
+    if llm is None:
+        st.error("Cannot proceed without a valid API key for the selected model.")
+    else:
+        with st.spinner('Processing...'):
+            try:
+                start_time = time.time()
+                results = crew.kickoff(inputs={"topic": patent_area, "stakeholder": stakeholder})
+                elapsed_time = time.time() - start_time
+
+                writer_output = getattr(results.tasks_output[2], "raw", "No details available.")
+                if writer_output:
+                    st.markdown("### Final Report")
+                    st.write(writer_output)
+                else:
+                    st.warning("No final report available.")
+
+                with st.expander("Explore Detailed Insights"):
+                    tab1, tab2 = st.tabs(["Planner's Insights", "Analyst's Analysis"])
+
+                    with tab1:
+                        planner_output = getattr(results.tasks_output[0], "raw", "No details available.")
+                        st.write(planner_output)
+
+                    with tab2:
+                        analyst_output = getattr(results.tasks_output[1], "raw", "No details available.")
+                        st.write(analyst_output)
+
+                        charts = []
+                        if enable_advanced_analysis:
+                            charts = create_visualizations(analyst_output)
+
+                        table_data = display_table(analyst_output)
+
+                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}
+                )
+                with open(pdf_path, "rb") as report_file:
+                    st.download_button("Download Report", data=report_file, file_name="Patent_Strategy_Report.pdf")
+
+            except Exception as e:
+                logging.error(f"An error occurred during execution: {e}")
+                st.error(f"An error occurred during execution: {e}")