Update app.py

app.py

@@ -69,22 +69,13 @@ st.write("---")
 # Sidebar for API key configuration
 with st.sidebar:
     st.title("⚙️ Configuration")
-    api_key_source = st.radio("Select API Key Provider:", 
-                            ["Google (Gemini)", "OpenAI"],
-                            help="Choose which AI provider to use")
     
-    if api_key_source == "Google (Gemini)":
-        api_key = st.text_input("Enter your Gemini API Key", type="password", 
-                                help="Required for the AI model to function")
-        if api_key:
-            os.environ["GEMINI_API_KEY"] = api_key
-            os.environ["GOOGLE_API_KEY"] = api_key
-    else:
-        api_key = st.text_input("Enter your OpenAI API Key", type="password", 
-                                help="Required for the AI model to function")
-        if api_key:
-            os.environ["OPENAI_API_KEY"] = api_key
-            
+    api_key = st.text_input("Enter your Gemini API Key", type="password", 
+                          help="Required for the AI model to function")
+    if api_key:
+        os.environ["GEMINI_API_KEY"] = api_key
+        os.environ["GOOGLE_API_KEY"] = api_key
+    
     st.divider()
     
     # Reset button
@@ -93,7 +84,6 @@ with st.sidebar:
             del st.session_state[key]
         st.rerun()
 
-
 #---------------------------- Utility Functions ----------------------------#
 
 def extract_text_from_pdf(file):
@@ -398,6 +388,7 @@ class CaseBreakdownCrew:
         self.api_key = api_key
         
     def create_metadata_agent(self):
+        llm = LLM(model='gemini/gemini-2.0-flash', api_key=self.api_key)
         return Agent(
             role="Metadata Analyzer",
             goal="Extract title and author information from document content",
@@ -409,6 +400,7 @@ class CaseBreakdownCrew:
         )
     
     def create_content_generator_agent(self):
+        llm = LLM(model='gemini/gemini-2.0-flash', api_key=self.api_key)
         return Agent(
             role="Case Study Content Generator",
             goal="Generate comprehensive case analysis content based on section requirements",
@@ -420,6 +412,7 @@ class CaseBreakdownCrew:
         )
     
     def create_content_reviewer_agent(self):
+        llm = LLM(model='gemini/gemini-2.0-flash', api_key=self.api_key)
         return Agent(
             role="Content Quality Reviewer",
             goal="Evaluate and score content for quality, relevance, and depth",
@@ -607,16 +600,10 @@ def create_teaching_plan_crew(file_paths, llm_provider="gemini"):
     tracker.set_placeholder(st.empty())
     
     # Initialize LLM based on provider
-    if llm_provider == "gemini":
-        my_llm = LLM(
-            model='gemini/gemini-2.0-flash',
-            api_key=os.environ.get("GEMINI_API_KEY")
-        )
-    else:
         my_llm = LLM(
-            model='gpt-4-turbo',
-            api_key=os.environ.get("OPENAI_API_KEY")
-        )
+        model='gemini/gemini-2.0-flash',
+        api_key=os.environ.get("GEMINI_API_KEY")
+    )
     
     # Define agents with callbacks for UI updates
     pdf_analyzer = Agent(
@@ -708,21 +695,15 @@ def create_teaching_plan_crew(file_paths, llm_provider="gemini"):
 #---------------------------- Board Plan Generator ----------------------------#
 
 class BoardPlanAnalyzer:
-    def __init__(self, llm_provider="gemini"):
-        if llm_provider == "gemini":
-            api_key = os.environ.get('GEMINI_API_KEY')
-            self.model = "gemini/gemini-2.0-flash"
-        else:
-            api_key = os.environ.get('OPENAI_API_KEY')
-            self.model = "gpt-4-turbo"
+    def __init__(self):
+        api_key = os.environ.get('GEMINI_API_KEY')
+        self.model = "gemini/gemini-2.0-flash"
             
         if not api_key:
-            raise ValueError(f"{llm_provider.capitalize()} API key not found")
+            raise ValueError("Gemini API key not found")
             
-        if llm_provider == "gemini":
-            os.environ['GEMINI_API_KEY'] = api_key
-        else:
-            os.environ['OPENAI_API_KEY'] = api_key
+        os.environ['GEMINI_API_KEY'] = api_key
+        self.api_key = api_key
             
         litellm.set_verbose = True
         
@@ -745,6 +726,7 @@ class BoardPlanAnalyzer:
                 description="Extracts text content from PDF files"
             )],
             allow_delegation=False,
+            llm=self.model,
             verbose=True
         )
 
@@ -761,6 +743,7 @@ class BoardPlanAnalyzer:
                 description="Analyzes case study and creates structured board plan"
             )],
             allow_delegation=False,
+            llm=self.model,
             verbose=True
         )
 
@@ -904,7 +887,7 @@ if st.session_state.uploaded_files:
                 st.warning("⚠️ Please enter an API key in the sidebar before proceeding.")
             else:
                 # Initialize the breakdown generator
-                crew_manager = CaseBreakdownCrew(api_key)
+                crew_manager = CaseBreakdownCrew(api_key, api_key_source)
                 
                 # Extract metadata
                 with st.spinner("Extracting document metadata..."):
@@ -1098,7 +1081,7 @@ if st.session_state.uploaded_files:
                         progress_bar = progress_placeholder.progress(0)
                         
                         # Select LLM provider
-                        llm_provider = "gemini" if api_key_source == "Google (Gemini)" else "openai"
+                        llm_provider = "gemini"
                         
                         # Update progress
                         progress_bar.progress(10)