Update app.py

app.py

@@ -7,7 +7,6 @@ from langchain_community.vectorstores import FAISS
 import requests
 import os
 import time
-import base64
 
 # Page configuration
 st.set_page_config(
@@ -124,10 +123,9 @@ def check_token_validity():
     except:
         return False
 
-def query_hf_inference_api(prompt, max_tokens=200):
+def query_hf_inference_api(prompt, max_tokens=200, model="google/flan-t5-base"):
     """Query Hugging Face Inference API with better error handling"""
-    MODEL = "google/flan-t5-base"  # Switch to base model for better accessibility
-    API_URL = f"https://api-inference.huggingface.co/models/{MODEL}"
+    API_URL = f"https://api-inference.huggingface.co/models/{model}"
     headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"} if os.getenv('HF_API_KEY') else {}
     
     payload = {
@@ -155,10 +153,10 @@ def query_hf_inference_api(prompt, max_tokens=200):
                 <h4>403 Forbidden Error</h4>
                 <p>Token is set: <strong>{'Yes' if os.getenv('HF_API_KEY') else 'No'}</strong></p>
                 <p>Token valid: <strong>{'Yes' if st.session_state.token_valid else 'No'}</strong></p>
-                <p>Model: {MODEL}</p>
+                <p>Model: {model}</p>
                 <p>Possible solutions:</p>
                 <ol>
-                    <li>Visit the <a href="https://huggingface.co/google/flan-t5-base" target="_blank">model page</a> and click "Agree and access repository"</li>
+                    <li>Visit the <a href="https://huggingface.co/{model}" target="_blank">model page</a> and click "Agree and access repository"</li>
                     <li>Ensure your token has "read" permissions</li>
                     <li>Wait 5-10 minutes after accepting terms</li>
                     <li>Try a different model using the dropdown below</li>
@@ -171,7 +169,7 @@ def query_hf_inference_api(prompt, max_tokens=200):
         elif response.status_code == 429:
             st.warning("Rate limit exceeded. Waiting and retrying...")
             time.sleep(3)
-            return query_hf_inference_api(prompt, max_tokens)
+            return query_hf_inference_api(prompt, max_tokens, model)
         
         else:
             st.error(f"API Error {response.status_code}: {response.text[:200]}")
@@ -304,7 +302,7 @@ with st.expander("🔧 Debug Information", expanded=False):
         <div class="info">
             <p>Your token is set but we're still having issues. Try these steps:</p>
             <ol>
-                <li>Visit the <a href="https://huggingface.co/google/flan-t5-base" target="_blank">model page</a></li>
+                <li>Visit the model page for your selected model</li>
                 <li>Click "Agree and access repository"</li>
                 <li>Wait 5-10 minutes for changes to propagate</li>
                 <li>Try a different model from the dropdown</li>
@@ -324,10 +322,15 @@ with st.expander("🔧 Debug Information", expanded=False):
         </div>
         """, unsafe_allow_html=True)
 
-# PDF Upload Section
+# PDF Upload Section (FIXED LABEL ERROR)
 with st.container():
     st.subheader("📤 Upload Your Textbook/Notes")
-    pdf_file = st.file_uploader("", type="pdf", label_visibility="collapsed")
+    # Fixed empty label issue by adding a space and hiding it
+    pdf_file = st.file_uploader(
+        "Upload PDF", 
+        type="pdf", 
+        label_visibility="collapsed"
+    )
 
 # Main content
 if pdf_file:
@@ -415,6 +418,6 @@ if pdf_file:
 st.markdown("---")
 st.markdown("""
 <div style="text-align: center; padding: 20px;">
-    Built with ❤️ for students | PDF Study Assistant v4.0
+    Built with ❤️ for students | PDF Study Assistant v4.1
 </div>
 """, unsafe_allow_html=True)