Update app.py

app.py

@@ -16,7 +16,25 @@ if not openai_api_key:
     st.stop()
 
 # Initialize the LLM
-llm = OpenAI(api_token=openai_api_key)
+#llm = OpenAI(api_token=openai_api_key)
+
+def initialize_llm(model_choice):
+    groq_api_key = os.getenv("GROQ_API_KEY")
+    openai_api_key = os.getenv("OPENAI_API_KEY")
+
+    if model_choice == "llama-3.3-70b":
+        if not groq_api_key:
+            st.error("Groq API key is missing. Please set the GROQ_API_KEY environment variable.")
+            return None
+        return ChatGroq(groq_api_key=groq_api_key, model="groq/llama-3.3-70b-versatile")
+    elif model_choice == "GPT-4o":
+        if not openai_api_key:
+            st.error("OpenAI API key is missing. Please set the OPENAI_API_KEY environment variable.")
+            return None
+        return ChatOpenAI(api_key=openai_api_key, model="gpt-4o")
+
+model_choice = st.radio("Select LLM", ["GPT-4o", "llama-3.3-70b"], index=0, horizontal=True)
+llm = initialize_llm(model_choice)
 
 
 def load_dataset_into_session():