Update app.py

app.py

@@ -9,7 +9,7 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
 def get_pdf_text(pdf_docs):
-    """Extracts text from uploaded PDF files"""
+    """Extracts text from uploaded PDF files."""
     text = ""
     for pdf in pdf_docs:
         pdf_reader = PdfReader(pdf)
@@ -18,14 +18,17 @@ def get_pdf_text(pdf_docs):
     return text
 
 def get_gemini_response(question, context):
-    """Sends question and context to Gemini for an answer"""
+    """Sends question and context to Gemini for an answer."""
     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)
+        response = genai.ChatCompletion.create(
+            model="gemini-pro",
+            messages=[{"role": "user", "content": prompt}],
+        )
         st.write(f"API Response:\n{response}")  # Debugging: Check raw response
-        if response and response.generations:
-            return response.generations[0].text
+        if response and response['choices']:
+            return response['choices'][0]['message']['content']
         else:
             return "No valid response from the API."
     except Exception as e: