Update app.py

app.py

@@ -2,13 +2,28 @@ import os
 import gradio as gr
 import google.generativeai as genai
 from dotenv import load_dotenv
+import httpx
 
 load_dotenv()
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Configure model with system prompt
+# PDF URL for course information
+COURSE_PDF_URL = "https://drive.google.com/file/d/1Xf46sT9kSzSVdekSCnz3NsBodpGET1Gi/view?usp=drive_link"
+
+# Load PDF content once at startup
+def get_pdf_content():
+    try:
+        # Convert Google Drive link to direct download link
+        file_id = COURSE_PDF_URL.split('/')[5]
+        direct_url = f"https://drive.google.com/uc?export=download&id={file_id}"
+        return httpx.get(direct_url).content
+    except Exception as e:
+        print(f"Error loading PDF: {e}")
+        return None
+
+# Configure model with system prompt and PDF content
 model = genai.GenerativeModel(
-    model_name="gemini-2.0-flash",
+    model_name="gemini-1.5-flash",
     generation_config={
         "temperature": 0.9,
         "top_p": 1,
@@ -16,6 +31,34 @@ model = genai.GenerativeModel(
     }
 )
 
+def chat(message, history):
+    try:
+        # Get PDF content
+        pdf_content = get_pdf_content()
+        
+        # Prepare the message with PDF context
+        if pdf_content:
+            messages = [
+                {"role": "user", "parts": [
+                    system_prompt,
+                    types.Part.from_bytes(data=pdf_content, mime_type='application/pdf'),
+                    "Use the PDF content to provide detailed answers about the course."
+                ]},
+                *[{"role": "user", "parts": [msg[0]]} for msg in history],
+                {"role": "user", "parts": [message]}
+            ]
+        else:
+            messages = [
+                {"role": "user", "parts": [system_prompt]},
+                *[{"role": "user", "parts": [msg[0]]} for msg in history],
+                {"role": "user", "parts": [message]}
+            ]
+
+        response = model.generate_content(messages)
+        return response.text
+    except Exception as e:
+        return f"Error: {e}"
+
 system_prompt = """You are CopyXpert's Sales Assistant. Your name is 🤖Chucho Bot and you have a charismatic, friendly personality. You ONLY talk about CopyXpert course.
 
 IF USERS ASK ANYTHING NOT RELATED TO COPYXPERT, respond with one of these phrases (vary them creatively):