Turbiling commited on
Commit
6715cbe
Β·
verified Β·
1 Parent(s): bbd6414

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -55,7 +55,6 @@ def process_file(file_path):
55
  async def start():
56
  await cl.Message("πŸ‘‹ Welcome! Please upload a PDF to get started.").send()
57
 
58
- # Let user upload file
59
  msg = await cl.AskFileMessage(
60
  content="πŸ“„ Upload your PDF file here:",
61
  accept=["application/pdf"],
@@ -67,10 +66,11 @@ async def start():
67
  file = msg.files[0]
68
  qa_chain = process_file(file.path)
69
 
70
- # Save QA chain to session
71
  cl.user_session.set("qa_chain", qa_chain)
72
  cl.user_session.set("ready", True)
73
  await cl.Message(f"βœ… File `{file.name}` processed. You can now ask your questions.").send()
 
 
74
 
75
  # On user message
76
  @cl.on_message
@@ -80,6 +80,9 @@ async def handle_message(message: cl.Message):
80
  return
81
 
82
  qa_chain = cl.user_session.get("qa_chain")
83
- response = qa_chain(message.content)
84
- answer = response["result"]
85
- await cl.Message(content=answer).send()
 
 
 
 
55
  async def start():
56
  await cl.Message("πŸ‘‹ Welcome! Please upload a PDF to get started.").send()
57
 
 
58
  msg = await cl.AskFileMessage(
59
  content="πŸ“„ Upload your PDF file here:",
60
  accept=["application/pdf"],
 
66
  file = msg.files[0]
67
  qa_chain = process_file(file.path)
68
 
 
69
  cl.user_session.set("qa_chain", qa_chain)
70
  cl.user_session.set("ready", True)
71
  await cl.Message(f"βœ… File `{file.name}` processed. You can now ask your questions.").send()
72
+ else:
73
+ await cl.Message("❌ No file was uploaded. Please try again.").send()
74
 
75
  # On user message
76
  @cl.on_message
 
80
  return
81
 
82
  qa_chain = cl.user_session.get("qa_chain")
83
+ try:
84
+ response = qa_chain(message.content)
85
+ answer = response["result"]
86
+ await cl.Message(content=answer).send()
87
+ except Exception as e:
88
+ await cl.Message(content=f"❌ An error occurred while generating a response: {str(e)}").send()