Update app.py

app.py

@@ -1,12 +1,12 @@
 import chainlit as cl
 import os
-import fitz  # PyMuPDF
+import fitz  # PyMuPDF for PDF reading
 from groq import Groq
 
-# Initialize Groq client
+# Initialize Groq client using your Hugging Face Secret
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# Function to extract text from PDF
+# Function to extract text from PDF file
 def extract_pdf_text(file_path):
     text = ""
     with fitz.open(file_path) as doc:
@@ -16,34 +16,35 @@ def extract_pdf_text(file_path):
 
 @cl.on_chat_start
 async def start_chat():
-    await cl.Message(content="👋 Welcome! Upload a PDF or ask me anything.").send()
+    await cl.Message(content="👋 Welcome! Upload a PDF or ask me any question.").send()
 
 @cl.on_message
 async def handle_message(message: cl.Message):
     try:
-        # ✅ Check if user uploaded a file
+        # If user uploaded a file (PDF)
         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()
+            file_path = file.path  # Access the uploaded file path directly
+            await cl.Message(content=f"📄 File uploaded: `{file.name}`").send()
 
-            # ✅ Extract text from PDF
+            # Extract PDF text
             pdf_text = extract_pdf_text(file_path)
 
-            # ✅ Ask Groq to summarize the content
+            # Ask Groq to summarize it
             response = groq_client.chat.completions.create(
                 model="llama3-8b-8192",
                 messages=[
-                    {"role": "system", "content": "Summarize the uploaded PDF."},
-                    {"role": "user", "content": pdf_text[:4000]}  # Limit to 4k tokens
+                    {"role": "system", "content": "Summarize the uploaded PDF content."},
+                    {"role": "user", "content": pdf_text[:4000]}  # limit to 4k tokens
                 ]
             )
 
             reply = response.choices[0].message.content
+            await cl.sleep(1)  # Fix for Chainlit session issue
             await cl.Message(content=f"📚 Summary:\n{reply}").send()
 
+        # If it's a normal question
         else:
-            # ✅ Handle normal text questions
             user_input = message.content
             await cl.Message(content=f"📝 Question: {user_input}").send()
 
@@ -56,8 +57,8 @@ async def handle_message(message: cl.Message):
             )
 
             reply = response.choices[0].message.content
+            await cl.sleep(1)  # Fix for Chainlit session issue
             await cl.Message(content=reply).send()
 
     except Exception as e:
         await cl.Message(content=f"❌ Error: {str(e)}").send()
-