Update app.py

app.py

@@ -19,9 +19,18 @@ def get_pdf_text(pdf_docs):
 
 def get_gemini_response(question, context):
     """Sends question and context to Gemini for an answer"""
-    prompt = f"Context:\n{context}\n\nQuestion:\n{question}\n\nAnswer:"
-    response = genai.generate_text(prompt=prompt)
-    return response.generations[0].text
+    try:
+        prompt = f"Context:\n{context}\n\nQuestion:\n{question}\n\nAnswer:"
+        st.write(f"Sending prompt to Gemini:\n{prompt}")  # Debugging: Check prompt
+        response = genai.generate_text(prompt=prompt)
+        st.write(f"API Response:\n{response}")  # Debugging: Check raw response
+        if response and response.generations:
+            return response.generations[0].text
+        else:
+            return "No valid response from the API."
+    except Exception as e:
+        st.error(f"Error generating response: {e}")
+        return "Error while generating response."
 
 def main():
     st.set_page_config(page_title="Chat with PDF")