Trying to fix connection error

config/settings.py

@@ -13,6 +13,11 @@ class Settings:
     
     # API Keys
     OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+    if not OPENAI_API_KEY:
+        raise ValueError(
+            "OpenAI API key not found. Please set OPENAI_API_KEY "
+            "environment variable."
+        )
     HF_TOKEN = os.getenv("HF_TOKEN")
     
     # Directory Paths

requirements.txt

@@ -6,6 +6,7 @@ chromadb>=0.4.22
 streamlit==1.41.0
 python-dotenv==1.0.1
 huggingface-hub>=0.19.4
+datasets>=2.16.0
 pytest>=7.4.0
 black>=23.0.0
 flake8>=6.0.0

src/generator/generator.py

@@ -201,11 +201,16 @@ class LyricGenerator:
             chat_history = []
 
         try:
+            print("Starting lyrics generation...")
+            print(f"OpenAI API Key present: {bool(Settings.OPENAI_API_KEY)}")
+            
             # Get source documents with scores first
+            print("Searching for similar documents...")
             docs_and_scores = self.vector_store.similarity_search_with_score(
                 prompt,
                 k=20
             )
+            print(f"Found {len(docs_and_scores)} similar documents")
 
             # Sort by similarity (convert distance to similarity)
             docs_and_scores.sort(key=lambda x: x[1], reverse=False)
@@ -218,15 +223,25 @@ class LyricGenerator:
                     'artist': doc.metadata['artist'],
                     'song': doc.metadata['song_title'],
                     'similarity': similarity,
-                    # First 200 chars
-                    'content': doc.page_content[:200] + "..."
+                    'content': doc.page_content[:200] + "..."  # First 200 chars
                 })
 
-            # Generate response using invoke
-            response = self.qa_chain.invoke({
-                "question": prompt,
-                "chat_history": chat_history
-            })
+            print("Generating response using QA chain...")
+            try:
+                # Generate response using invoke
+                response = self.qa_chain.invoke({
+                    "question": prompt,
+                    "chat_history": chat_history
+                })
+                print("Successfully generated response")
+            except Exception as e:
+                print(f"Error in QA chain: {str(e)}")
+                if "Connection" in str(e):
+                    raise RuntimeError(
+                        "OpenAI API connection error. Please check your internet "
+                        "connection and API key."
+                    )
+                raise
 
             # Add detailed context to response
             response["source_documents_with_scores"] = docs_and_scores
@@ -235,4 +250,5 @@ class LyricGenerator:
             return response
 
         except Exception as e:
+            print(f"Error in generate_lyrics: {str(e)}")
             raise RuntimeError(f"Failed to generate lyrics: {str(e)}")