Update app.py

app.py

@@ -378,12 +378,27 @@ Please provide a thorough, well-structured answer that directly addresses the qu
                 if response.status_code == 200:
                     ai_response = response.json()['choices'][0]['message']['content'].strip()
                     ai_answer = ai_response if len(ai_response) > 10 else extracted_answer
-                elif response.status_code == 402:
-                    st.warning("💳 OpenRouter credits exhausted. Using extracted answers only.")
-                elif response.status_code == 429:
-                    st.warning("⏱️ Rate limit reached. Using extracted answers only.")
                 else:
-                    st.warning(f"API Error {response.status_code}. Using extracted answers only.")
+                    # Log the actual error for debugging
+                    error_detail = ""
+                    try:
+                        error_detail = response.json().get('error', {}).get('message', '')
+                    except:
+                        pass
+                    
+                    if response.status_code == 402:
+                        st.warning("💳 OpenRouter credits exhausted. Using extracted answers only.")
+                    elif response.status_code == 429:
+                        st.warning("⏱️ Rate limit reached. Using extracted answers only.")
+                    elif response.status_code == 401:
+                        st.error("🔑 Invalid API key. Check your OpenRouter key.")
+                    elif response.status_code == 400:
+                        st.error(f"❌ Bad request: {error_detail}")
+                    else:
+                        st.warning(f"API Error {response.status_code}: {error_detail}. Using extracted answers only.")
+                    
+                    # Debug info for troubleshooting
+                    st.info(f"Debug: Status {response.status_code}, Error: {error_detail}")
                     
             except requests.exceptions.Timeout:
                 st.warning("⏱️ API timeout. Using extracted answers only.")
@@ -435,12 +450,24 @@ def get_general_ai_response(query, unlimited_tokens=False):
         
         if response.status_code == 200:
             return response.json()['choices'][0]['message']['content'].strip()
-        elif response.status_code == 402:
-            return "Sorry, OpenRouter credits exhausted. Please add more credits or ask document-specific questions."
-        elif response.status_code == 429:
-            return "Rate limit reached. Please try again in a moment."
         else:
-            return f"API error (Status: {response.status_code}). Try asking about documents instead."
+            # Get detailed error information
+            error_detail = ""
+            try:
+                error_detail = response.json().get('error', {}).get('message', '')
+            except:
+                pass
+            
+            if response.status_code == 402:
+                return "Sorry, OpenRouter credits exhausted. Please add more credits or top up your account."
+            elif response.status_code == 429:
+                return "Rate limit reached. Please try again in a moment."
+            elif response.status_code == 401:
+                return "Invalid API key. Please check your OpenRouter API key configuration."
+            elif response.status_code == 400:
+                return f"Bad request: {error_detail}. Please try rephrasing your question."
+            else:
+                return f"API error (Status: {response.status_code}): {error_detail}. Please try again."
             
     except requests.exceptions.Timeout:
         return "Request timeout. Please try again."
@@ -721,8 +748,8 @@ if prompt := st.chat_input("Ask questions about your documents..."):
             # Search documents first
             search_results = rag_system.search(prompt, n_results=3)
             
-            # Check if we found relevant documents (much lower threshold)
-            if search_results and search_results[0]['similarity'] > 0.01:  # Very low threshold
+            # Check if we found relevant documents (lower threshold for better coverage)
+            if search_results and search_results[0]['similarity'] > 0.005:  # Even lower threshold
                 # Generate document-based answer
                 result = rag_system.generate_answer(
                     prompt, 
@@ -735,9 +762,18 @@ if prompt := st.chat_input("Ask questions about your documents..."):
                 if use_ai_enhancement and result['has_both']:
                     answer_text = result['ai_answer']
                     st.markdown(f"🤖 **AI Enhanced Answer:** {answer_text}")
+                    
+                    # Also show extracted answer for comparison if different
+                    if result['extracted_answer'] != answer_text:
+                        with st.expander("📄 View Extracted Answer"):
+                            st.markdown(result['extracted_answer'])
                 else:
                     answer_text = result['extracted_answer']
                     st.markdown(f"📄 **Document Answer:** {answer_text}")
+                    
+                    # Show why AI enhancement wasn't used
+                    if use_ai_enhancement and not result['has_both']:
+                        st.info("💡 AI enhancement failed - showing extracted answer from documents")
                 
                 # Show RAG info
                 if show_sources and result['sources']: