Update app.py

Added: max_retries=3, # <--- Add this line
timeout=30 # <--- Add this line
answer = response.get("answer", "")
return str(answer) if answer else "The model could not find an answer in that document."

show_error=True

app.py

@@ -21,7 +21,9 @@ def get_llm():
         task="text-generation",
         max_new_tokens=512,
         temperature=0.1,
-        huggingfacehub_api_token=os.getenv("HF_TOKEN")
+        huggingfacehub_api_token=os.getenv("HF_TOKEN");
+        max_retries=3, 
+        timeout=30     
     )
 
 # 2. DOCUMENT PROCESSING
@@ -49,6 +51,8 @@ def process_pdf(file_path):
 def retriever_qa(file, query):
     if file is None:
         return "Please upload a PDF first."
+    if not query:
+        return "Error: Please enter a question."
         
     llm = get_llm()
     retriever_obj = process_pdf(file)
@@ -69,7 +73,8 @@ def retriever_qa(file, query):
 
     # Invoke and return result
     response = rag_chain.invoke({"input": query})
-    return response["answer"] # Note: 'answer' instead of 'result'
+    answer = response.get("answer", "")
+    return str(answer) if answer else "The model could not find an answer in that document."
 
 # 4. GRADIO INTERFACE
 rag_application = gr.Interface(
@@ -83,4 +88,4 @@ rag_application = gr.Interface(
     title='Granite PDF Chatbot'
 )
 
-rag_application.launch()
+rag_application.launch(show_error=True)