Update app.py

app.py

@@ -11,14 +11,22 @@ sys.path.append(os.path.join(os.path.dirname(os.path.abspath(__file__)), "src"))
 from dotenv import load_dotenv
 load_dotenv()
 
-# Check API key BEFORE importing rag (which needs it)
+# Debug: Show if API key is loaded
 api_key = os.getenv("GOOGLE_API_KEY")
 if not api_key:
     st.error("⚠️ **GOOGLE_API_KEY is missing!**")
-    st.info("Please add your Google API key to the Secrets in your Hugging Face Space settings, or create a `.env` file with `GOOGLE_API_KEY=your_key_here`")
+    st.info("👉 Go to your HF Space **Settings → Secrets** and add `GOOGLE_API_KEY` with your Google AI Studio API key.")
     st.stop()
 
-from src.rag import get_answer
+# Try importing rag and show any import errors
+try:
+    from src.rag import get_answer
+    st.success("✅ RAG module loaded successfully!")
+except Exception as e:
+    st.error(f"❌ Failed to import RAG module: {e}")
+    import traceback
+    st.code(traceback.format_exc())
+    st.stop()
 
 # Title
 st.title("🌟 Sagar's Personal Assistant")
@@ -47,14 +55,10 @@ if prompt := st.chat_input("What would you like to know?"):
                 response = get_answer(prompt)
                 if response:
                     st.markdown(response)
-                    # Add assistant message to history
                     st.session_state.messages.append({"role": "assistant", "content": response})
                 else:
-                    st.error("❌ Received empty response from the AI. Please try again.")
+                    st.error("❌ Received empty response from AI.")
             except Exception as e:
-                error_msg = f"❌ An error occurred: {str(e)}"
-                st.error(error_msg)
-                # Also log to console for debugging
-                print(f"Error in get_answer: {e}")
+                st.error(f"❌ Error: {str(e)}")
                 import traceback
-                traceback.print_exc()
+                st.code(traceback.format_exc())