Update app.py

app.py

@@ -14,12 +14,11 @@ import faiss
 import uuid
 from dotenv import load_dotenv
 
-# Load environment variables
 load_dotenv()
 HUGGINGFACEHUB_API_TOKEN = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 RAG_ACCESS_KEY = os.getenv("RAG_ACCESS_KEY")
 
-# Initialize session state
+# Session state init
 if "vectorstore" not in st.session_state:
     st.session_state.vectorstore = None
 if "history" not in st.session_state:
@@ -27,7 +26,7 @@ if "history" not in st.session_state:
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
 
-# Sidebar with BSNL logo and authentication
+# Sidebar with login
 with st.sidebar:
     try:
         st.image("bsnl_logo.png", width=200)
@@ -37,7 +36,6 @@ with st.sidebar:
     st.header("RAG Control Panel")
     api_key_input = st.text_input("Enter RAG Access Key", type="password")
 
-    # Custom style for Authenticate button
     st.markdown("""
         <style>
         .auth-button button {
@@ -83,7 +81,7 @@ with st.sidebar:
         st.write(f"**A{i+1}:** {a}")
         st.markdown("---")
 
-# Main app UI
+# Main UI
 def main():
     st.markdown("""
         <style>
@@ -116,10 +114,10 @@ def main():
             except Exception as e:
                 st.error(f"Error generating answer: {str(e)}")
 
-# PDF processing logic
+# Processing PDF
 def process_input(input_data):
     os.makedirs("vectorstore", exist_ok=True)
-    os.chmod("vectorstore", 0o777)
+    os.chmod("vectorstore", 0o700)  # safer permissions
 
     progress_bar = st.progress(0)
     status = st.status("Processing PDF file...", expanded=True)
@@ -163,20 +161,20 @@ def process_input(input_data):
     progress_bar.progress(1.0)
     return vector_store
 
-# Question-answering logic
+# Q&A logic
 def answer_question(vectorstore, query):
     try:
         llm = HuggingFaceHub(
             repo_id="mistralai/Mistral-7B-Instruct-v0.1",
-            model_kwargs={"temperature": 0.7, "max_length": 512},
+            model_kwargs={"temperature": 0.7, "max_new_tokens": 512},
             huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN
         )
     except Exception as e:
-        raise RuntimeError("Failed to load LLM. Check Hugging Face API key and access rights.") from e
+        raise RuntimeError("Failed to load LLM. Ensure your Hugging Face API key is valid and the model is public or accessible.") from e
 
     retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
     prompt_template = PromptTemplate(
-        template="Use the context to answer the question concisely:\n\nContext: {context}\n\nQuestion: {question}\n\nAnswer:",
+        template="Use the context below to answer the question.\n\nContext: {context}\n\nQuestion: {question}\n\nAnswer:",
         input_variables=["context", "question"]
     )
 
@@ -191,6 +189,5 @@ def answer_question(vectorstore, query):
     result = qa_chain({"query": query})
     return result["result"].split("Answer:")[-1].strip()
 
-# Run the app
 if __name__ == "__main__":
     main()