Update app.py

app.py

@@ -3,10 +3,10 @@ import chainlit as cl
 from groq import Groq
 import fitz  # PyMuPDF
 
-# ✅ Initialize Groq Client
+# ✅ Initialize Groq client
 groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# ✅ Global store for uploaded PDF
+# ✅ Global variable for 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
 
-    # ✅ Check for uploaded PDF
+    # ✅ Handle uploaded PDF
     if message.elements:
         for element in message.elements:
             if element.name.endswith(".pdf"):
@@ -26,25 +26,27 @@ async def on_message(message: cl.Message):
                     text = ""
                     for page in doc:
                         text += page.get_text()
-                    pdf_context = text[:4000]  # limit content
-                    await cl.Message(content="✅ PDF uploaded! Now ask a question.").send()
+                    pdf_context = text[:4000]  # Limit size
+                    await cl.Message(content="✅ PDF uploaded! Now ask a question based on it.").send()
                     return
                 except Exception as e:
                     await cl.Message(content=f"❌ Error reading PDF: {e}").send()
                     return
 
-    # ✅ Check for user message
-    if not message.content.strip():
+    # ✅ Handle text questions
+    prompt = message.content.strip()
+    if not prompt:
         await cl.Message(content="⚠️ Please enter a question.").send()
         return
 
-    # ✅ Prepare prompt
-    prompt = message.content.strip()
+    # ✅ Append PDF context if available
     if pdf_context:
         prompt = f"Use the following PDF content to answer the question:\n\n{pdf_context}\n\nQuestion: {prompt}"
+    else:
+        prompt = f"Answer the question:\n\n{prompt}"
 
     try:
-        # ✅ Send to Groq
+        # ✅ Call Groq API
         response = groq_client.chat.completions.create(
             model="llama3-8b-8192",
             messages=[
@@ -53,9 +55,16 @@ async def on_message(message: cl.Message):
             ]
         )
 
-        # ✅ Extract and send response
-        answer = response.choices[0].message.content.strip()
-        await cl.Message(content=answer).send()
+        # ✅ Print raw response to logs for debugging
+        print("=== Groq Raw Response ===")
+        print(response)
+
+        # ✅ Extract and display answer
+        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()
 
     except Exception as e:
         await cl.Message(content=f"❌ Error: {e}").send()