Update app.py

app.py

@@ -3,6 +3,7 @@ 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
@@ -24,6 +25,34 @@ 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)
@@ -51,9 +80,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(
     role="Patent Research Consultant",