Update app.py

app.py

@@ -3,7 +3,6 @@ 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
@@ -25,33 +24,6 @@ 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")
 
-# Initialize LLM
-llm = None
-
-# Model Selection
-#st.header("Model Selection")
-model_choice = st.radio("Select LLM", ["OpenAI Model", "Groq-based LLM"], index=0, horizontal=True)
-
-
-# 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")
 enable_advanced_analysis = st.sidebar.checkbox("Enable Advanced Analysis", value=True)
@@ -79,6 +51,8 @@ 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(
@@ -133,7 +107,7 @@ write = Task(
         "2. Include key findings, visual aids, and actionable strategies.\n"
         "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."
+        "5. Limit the document to 600 words."
     ),
     expected_output="A polished, stakeholder-ready patent insights document with actionable recommendations.",
     agent=writer
@@ -259,51 +233,42 @@ def display_table(analyst_output):
     return table_data
 
 # Main Execution Block
-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.subheader("Raw Analyst Output")
-                        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}")
+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_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}")