Update app.py

app.py

@@ -1,52 +1,48 @@
 import os
 import chainlit as cl
 from groq import Groq
-import fitz  # PyMuPDF for PDF reading
+import fitz  # For reading PDFs
 
-# Set up the Groq client with your API key
+# Initialize Groq client
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# To store uploaded PDF content
-uploaded_pdf_text = ""
+# Store uploaded PDF content
+pdf_context = ""
 
 @cl.on_chat_start
-async def on_chat_start():
-    await cl.Message(content="👋 Welcome! Upload a PDF or ask me anything.").send()
+async def start():
+    await cl.Message(content="👋 Welcome! Upload a PDF or ask a question.").send()
 
 @cl.on_message
-async def on_message(message: cl.Message):
-    global uploaded_pdf_text
-    user_input = message.content.strip()
-
-    # If a PDF was uploaded earlier, combine its content with the user query
-    full_prompt = user_input
-    if uploaded_pdf_text:
-        full_prompt = f"Use this PDF content to answer: {user_input}\n\n{uploaded_pdf_text}"
+async def main(message: cl.Message):
+    global pdf_context
+
+    # ✅ Handle PDF upload (if any)
+    if message.elements:
+        for element in message.elements:
+            if element.name.endswith(".pdf"):
+                doc = fitz.open(element.path)
+                text = ""
+                for page in doc:
+                    text += page.get_text()
+                pdf_context = text[:4000]  # Limit content
+                await cl.Message(content="✅ PDF uploaded! Now ask questions based on it.").send()
+                return
+
+    # ✅ Handle text questions
+    prompt = message.content.strip()
+    if pdf_context:
+        prompt = f"Use the following PDF content to answer the question:\n\n{pdf_context}\n\nQuestion: {prompt}"
 
     try:
         response = groq_client.chat.completions.create(
             model="llama3-8b-8192",
             messages=[
                 {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content": full_prompt}
+                {"role": "user", "content": prompt}
             ]
         )
-        answer = response.choices[0].message.content
-        await cl.Message(content=answer).send()
+        await cl.Message(content=response.choices[0].message.content).send()
 
     except Exception as e:
         await cl.Message(content=f"❌ Error: {str(e)}").send()
-
-@cl.on_file_upload
-async def on_file(file: cl.UploadedFile):
-    global uploaded_pdf_text
-    try:
-        file_path = file.path  # Chainlit already saves file and gives safe path
-        doc = fitz.open(file_path)
-        text = ""
-        for page in doc:
-            text += page.get_text()
-        uploaded_pdf_text = text[:4000]  # Limit to 4000 characters
-        await cl.Message(content="✅ PDF uploaded and processed! Now ask questions about it.").send()
-    except Exception as e:
-        await cl.Message(content=f"❌ Failed to read PDF: {str(e)}").send()