Update app.py

app.py

@@ -6,7 +6,7 @@ import fitz  # PyMuPDF
 # ✅ Initialize Groq client
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# ✅ Global variable for PDF content
+# ✅ Global variable to store PDF content
 pdf_context = ""
 
 @cl.on_chat_start
@@ -17,7 +17,7 @@ async def on_chat_start():
 async def on_message(message: cl.Message):
     global pdf_context
 
-    # ✅ Handle uploaded PDF
+    # ✅ Check for uploaded PDF
     if message.elements:
         for element in message.elements:
             if element.name.endswith(".pdf"):
@@ -26,27 +26,27 @@ async def on_message(message: cl.Message):
                     text = ""
                     for page in doc:
                         text += page.get_text()
-                    pdf_context = text[:4000]  # Limit size
-                    await cl.Message(content="✅ PDF uploaded! Now ask a question based on it.").send()
+                    pdf_context = text[:4000]  # Limit context to 4000 characters
+                    await cl.Message(content="✅ PDF uploaded! Now ask a question.").send()
                     return
                 except Exception as e:
                     await cl.Message(content=f"❌ Error reading PDF: {e}").send()
                     return
 
-    # ✅ Handle text questions
-    prompt = message.content.strip()
-    if not prompt:
+    # ✅ If no question was asked
+    if not message.content.strip():
         await cl.Message(content="⚠️ Please enter a question.").send()
         return
 
-    # ✅ Append PDF context if available
+    # ✅ Construct prompt
+    user_question = message.content.strip()
     if pdf_context:
-        prompt = f"Use the following PDF content to answer the question:\n\n{pdf_context}\n\nQuestion: {prompt}"
+        prompt = f"Use the following PDF content to answer the question:\n\n{pdf_context}\n\nQuestion: {user_question}"
     else:
-        prompt = f"Answer the question:\n\n{prompt}"
+        prompt = user_question
 
     try:
-        # ✅ Call Groq API
+        # ✅ Send request to Groq
         response = groq_client.chat.completions.create(
             model="llama3-8b-8192",
             messages=[
@@ -55,16 +55,18 @@ async def on_message(message: cl.Message):
             ]
         )
 
-        # ✅ Print raw response to logs for debugging
+        # ✅ Log raw response for debugging
         print("=== Groq Raw Response ===")
         print(response)
 
-        # ✅ Extract and display answer
+        # ✅ Extract assistant reply safely
         if response.choices and response.choices[0].message and response.choices[0].message.content:
             answer = response.choices[0].message.content.strip()
-            await cl.Message(content=answer).send()
         else:
-            await cl.Message(content="🤖 No response from model.").send()
+            answer = "🤖 No response received from model."
+
+        # ✅ Show in UI
+        await cl.Message(content=answer).send()
 
     except Exception as e:
-        await cl.Message(content=f"❌ Error: {e}").send()
+        await cl.Message(content=f"❌ Error: {str(e)}").send()