Upload 10 files

app/chatbot.py

@@ -5,19 +5,15 @@ from typing import Dict, Any, List, Optional
 from datetime import datetime
 from app.models import ChatMessage, ChatSession
 from app.rag_system import RAGSystem
-from dotenv import load_dotenv
-
-# Load environment variables
-load_dotenv()
 
 logger = logging.getLogger(__name__)
 
 class RAGChatbot:
     """RAG-powered chatbot with memory of PDF and lecture content"""
     
-    def __init__(self):
-        self.client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY", ""))
-        self.rag_system = RAGSystem()
+    def __init__(self, openai_api_key: str):
+        self.client = openai.OpenAI(api_key=openai_api_key)
+        self.rag_system = RAGSystem(openai_api_key=openai_api_key)
         self.sessions: Dict[str, ChatSession] = {}
         self.max_context_length = 8000  # Token limit for context
     
@@ -253,3 +249,7 @@ Always strive to be helpful while being honest about the limitations of your kno
         except Exception as e:
             logger.error(f"Failed to update session content {session_id}: {str(e)}")
             return False
+        
+    def set_api_key(self, api_key: str):
+        """Set the OpenAI API key dynamically."""
+        self.client = openai.OpenAI(api_key=api_key)

app/gradio_interface.py

@@ -14,10 +14,11 @@ from app.chatbot import RAGChatbot
 logger = logging.getLogger(__name__)
 
 # Initialize components
+openai_api_key = os.getenv("OPENAI_API_KEY", "")
 pdf_processor = PDFProcessor()
 lecture_generator = LectureGenerator()
-voice_synthesizer = VoiceSynthesizer()
-chatbot = RAGChatbot()
+voice_synthesizer = VoiceSynthesizer(openai_api_key=openai_api_key)
+chatbot = RAGChatbot(openai_api_key=openai_api_key)
 
 # Global state for sessions
 current_session = None
@@ -60,6 +61,15 @@ def create_gradio_interface():
         
         # Session state
         session_id_state = gr.State(value=str(uuid.uuid4()))
+        openai_key_state = gr.State(value="")
+        
+        with gr.Tab("🔑 API Key Setup"):
+            openai_key_input = gr.Textbox(
+                label="OpenAI API Key",
+                placeholder="Enter your OpenAI API key here",
+                type="password"
+            )
+            save_key_btn = gr.Button("Save API Key")
         
         with gr.Tab("📄 PDF Upload & Processing"):
             with gr.Row():
@@ -138,10 +148,13 @@ def create_gradio_interface():
                     refresh_stats_btn = gr.Button("Refresh Stats")
         
         # Event handlers
-        def process_pdf_handler(pdf_file, style, examples, learning_objectives, session_id):
+        def process_pdf_handler(pdf_file, style, examples, learning_objectives, session_id, openai_key):
             """Handle PDF processing or topic-based lecture generation"""
             global session_data
-            
+
+            # Pass the OpenAI key to the chatbot or other components
+            chatbot.set_api_key(openai_key)
+
             try:
                 # Check if we have either PDF or learning objectives
                 if pdf_file is None and not learning_objectives.strip():
@@ -304,12 +317,15 @@ def create_gradio_interface():
                 logger.error(f"Audio generation error: {str(e)}")
                 return None
         
-        def chat_handler(message, history, session_id):
+        def chat_handler(message, history, session_id, openai_key):
             """Handle chat messages"""
             if not message.strip():
                 return history, ""
             
             try:
+                # Pass the OpenAI key to the chatbot
+                chatbot.set_api_key(openai_key)
+
                 response_result = chatbot.get_response(session_id, message)
                 
                 if response_result['success']:
@@ -346,10 +362,20 @@ def create_gradio_interface():
             """Get chat statistics"""
             return chatbot.get_session_stats(session_id)
         
+        def save_openai_key(key):
+            """Save the OpenAI API key to the session state"""
+            return key
+        
         # Wire up event handlers
+        save_key_btn.click(
+            fn=save_openai_key,
+            inputs=[openai_key_input],
+            outputs=[openai_key_state]
+        )
+        
         process_btn.click(
             fn=process_pdf_handler,
-            inputs=[pdf_upload, lecture_style, include_examples, learning_objectives, session_id_state],
+            inputs=[pdf_upload, lecture_style, include_examples, learning_objectives, session_id_state, openai_key_state],
             outputs=[processing_status, pdf_info, session_id_state]
         ).then(
             fn=update_lecture_display,
@@ -371,13 +397,13 @@ def create_gradio_interface():
         
         send_btn.click(
             fn=chat_handler,
-            inputs=[msg_input, chatbot_interface, session_id_state],
+            inputs=[msg_input, chatbot_interface, session_id_state, openai_key_state],
             outputs=[chatbot_interface, msg_input]
         )
         
         msg_input.submit(
             fn=chat_handler,
-            inputs=[msg_input, chatbot_interface, session_id_state],
+            inputs=[msg_input, chatbot_interface, session_id_state, openai_key_state],
             outputs=[chatbot_interface, msg_input]
         )
         

app/lecture_generator.py

@@ -11,15 +11,11 @@ from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, PageBreak
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.lib.units import inch
 from reportlab.lib.colors import HexColor
-from dotenv import load_dotenv
-
-# Load environment variables
-load_dotenv()
 
 logger = logging.getLogger(__name__)
 
 # Set up OpenAI client
-openai.api_key = os.getenv("OPENAI_API_KEY", "")
+openai.api_key = None  # Placeholder, should be set externally
 
 class LectureState(TypedDict):
     pdf_content: str

app/rag_system.py

@@ -7,27 +7,20 @@ from typing import List, Dict, Any, Optional
 import uuid
 from datetime import datetime
 import numpy as np
-from dotenv import load_dotenv
-
-# Load environment variables
-load_dotenv()
 
 logger = logging.getLogger(__name__)
 
 class RAGSystem:
     """Retrieval-Augmented Generation system for chatbot functionality"""
     
-    def __init__(self, persist_directory: str = "chroma_db"):
-        self.client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY", ""))
+    def __init__(self, openai_api_key: str, persist_directory: str = "chroma_db"):
+        self.client = openai.OpenAI(api_key=openai_api_key)
         
         # Initialize ChromaDB
         self.chroma_client = chromadb.PersistentClient(path=persist_directory)
         
         # Create embedding function
-        self.embedding_function = embedding_functions.OpenAIEmbeddingFunction(
-            api_key=os.getenv("OPENAI_API_KEY", ""),
-            model_name="text-embedding-ada-002"
-        )
+        self.embedding_function = embedding_functions.DefaultEmbeddingFunction()
         
         # Collections for different document types
         self.pdf_collection = self._get_or_create_collection("pdf_documents")

app/utils.py

@@ -4,7 +4,6 @@ import hashlib
 from typing import Dict, Any, Optional
 from datetime import datetime
 import json
-from dotenv import load_dotenv
 
 def setup_logging():
     """Setup logging configuration"""
@@ -156,9 +155,6 @@ def measure_execution_time(func):
     
     return wrapper
 
-# Load environment variables
-load_dotenv()
-
 # Initialize logging when module is imported
 setup_logging()
 

app/voice_synthesizer.py

@@ -5,23 +5,23 @@ from typing import Dict, Any, Optional
 from pathlib import Path
 import tempfile
 import io
-from dotenv import load_dotenv
-
-# Load environment variables
-load_dotenv()
 
 logger = logging.getLogger(__name__)
 
 class VoiceSynthesizer:
     """Handles text-to-speech conversion for lecture content"""
     
-    def __init__(self):
-        self.client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY", ""))
+    def __init__(self, openai_api_key: str):
+        self.client = openai.OpenAI(api_key=openai_api_key)
         self.supported_voices = [
             "alloy", "echo", "fable", "onyx", "nova", "shimmer"
         ]
         self.default_voice = "nova"
     
+    def set_api_key(self, api_key: str):
+        """Set the OpenAI API key dynamically."""
+        self.client = openai.OpenAI(api_key=api_key)
+    
     def synthesize_lecture(self, lecture_content: str, voice: str = None, output_path: str = None) -> Dict[str, Any]:
         """
         Convert lecture text to speech using OpenAI TTS