Update app.py

app.py

@@ -1,31 +1,59 @@
 import chainlit as cl
 import os
+import fitz  # PyMuPDF
 from groq import Groq
 
+# Initialize Groq client
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
+# Extract text from uploaded PDF
+def extract_pdf_text(file_path):
+    text = ""
+    with fitz.open(file_path) as doc:
+        for page in doc:
+            text += page.get_text()
+    return text
+
 @cl.on_chat_start
 async def start_chat():
-    await cl.Message(content="👋 Hello! I am your Groq-powered chatbot.").send()
+    await cl.Message(content="👋 Upload a PDF or ask me anything!").send()
 
 @cl.on_message
 async def handle_message(message: cl.Message):
-    print("📥 Received message:", message.content)  # Debug log
+    # If user uploaded a file
+    if message.elements:
+        file = message.elements[0]
+        file_path = await file.download()
+        await cl.Message(content=f"📄 Received file: {file.name}").send()
 
-    try:
-        response = groq_client.chat.completions.create(
-            model="llama3-8b-8192",
-            messages=[
-                {"role": "system", "content": "You are a helpful assistant."},
-                {"role": "user", "content": message.content}
-            ]
-        )
-        reply = response.choices[0].message.content
-        print("📤 Groq response:", reply)  # Debug log
-        await cl.Message(content=reply).send()
+        try:
+            # Extract text
+            pdf_text = extract_pdf_text(file_path)
+            # Ask Groq a question about the content
+            response = groq_client.chat.completions.create(
+                model="llama3-8b-8192",
+                messages=[
+                    {"role": "system", "content": "You are a PDF-reading assistant. Answer questions based on the document."},
+                    {"role": "user", "content": f"Summarize this PDF:\n{pdf_text[:4000]}"}
+                ]
+            )
+            reply = response.choices[0].message.content
+            await cl.Message(content=f"🧠 Summary:\n{reply}").send()
 
-    except Exception as e:
-        error_msg = f"❌ Error: {str(e)}"
-        print(error_msg)
-        await cl.Message(content=error_msg).send()
-        
+        except Exception as e:
+            await cl.Message(content=f"❌ PDF error: {str(e)}").send()
+    
+    # If user typed a question
+    else:
+        try:
+            response = groq_client.chat.completions.create(
+                model="llama3-8b-8192",
+                messages=[
+                    {"role": "system", "content": "You are a helpful assistant."},
+                    {"role": "user", "content": message.content}
+                ]
+            )
+            reply = response.choices[0].message.content
+            await cl.Message(content=reply).send()
+        except Exception as e:
+            await cl.Message(content=f"❌ Error: {str(e)}").send()