Update app.py

app.py

@@ -6,7 +6,7 @@ from groq import Groq
 # Initialize Groq client
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# Extract text from uploaded PDF
+# Function to extract text from PDF
 def extract_pdf_text(file_path):
     text = ""
     with fitz.open(file_path) as doc:
@@ -16,44 +16,47 @@ def extract_pdf_text(file_path):
 
 @cl.on_chat_start
 async def start_chat():
-    await cl.Message(content="👋 Upload a PDF or ask me anything!").send()
+    await cl.Message(content="👋 Welcome! Upload a PDF or ask me anything.").send()
 
 @cl.on_message
 async def handle_message(message: cl.Message):
-    # 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:
-            # Extract text
+    try:
+        # ✅ Check if user uploaded a file
+        if message.elements:
+            file = message.elements[0]
+            file_path = file.path  # ✅ Use `file.path` directly — no download
+            await cl.Message(content=f"📄 File uploaded: {file.name}").send()
+
+            # ✅ Extract text from PDF
             pdf_text = extract_pdf_text(file_path)
-            # Ask Groq a question about the content
+
+            # ✅ Ask Groq to summarize 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]}"}
+                    {"role": "system", "content": "Summarize the uploaded PDF."},
+                    {"role": "user", "content": pdf_text[:4000]}  # Limit to 4k tokens
                 ]
             )
+
             reply = response.choices[0].message.content
-            await cl.Message(content=f"🧠 Summary:\n{reply}").send()
-
-        except Exception as e:
-            await cl.Message(content=f"❌ PDF error: {str(e)}").send()
-    
-    # If user typed a question
-    else:
-        try:
+            await cl.Message(content=f"📚 Summary:\n{reply}").send()
+
+        else:
+            # ✅ Handle normal text questions
+            user_input = message.content
+            await cl.Message(content=f"📝 Question: {user_input}").send()
+
             response = groq_client.chat.completions.create(
                 model="llama3-8b-8192",
                 messages=[
                     {"role": "system", "content": "You are a helpful assistant."},
-                    {"role": "user", "content": message.content}
+                    {"role": "user", "content": user_input}
                 ]
             )
+
             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()
+
+    except Exception as e:
+        await cl.Message(content=f"❌ Error: {str(e)}").send()