Update app.py

app.py

@@ -51,16 +51,21 @@ def chatbot(pdf_file, user_question):
     # Use only the first chunk (to optimize token usage)
     prompt = f"Based on this document, answer the question:\n\nDocument:\n{chunks[0]}\n\nQuestion: {user_question}"
 
+    try:
+            response = together.chat.completions.create(
+                model="mistralai/Mistral-7B-Instruct-v0.1",
+                prompt=prompt,
+                max_tokens=200,
+                temperature=0.7,
+            )
+        
+            # Return chatbot's response
+            return response.choices[0].message.content.strip()
+    except  Exception as e:
+        return f"Error generating response: {e}"
+        
     # Send to Together.AI (Mistral-7B)
-    response = together.Completion.create(
-        model="mistralai/Mistral-7B-Instruct-v0.1",
-        prompt=prompt,
-        max_tokens=200,
-        temperature=0.7,
-    )
 
-    # Return chatbot's response
-    return response.choices[0].text
 
 
 # Gradio Interface