BGarida commited on
Commit
a294dd2
·
1 Parent(s): d4f700d

Deploying PDF Version v9

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -72,7 +72,9 @@ def process_file(file: AskFileResponse):
72
 
73
 
74
  @cl.on_chat_start
75
- async def on_chat_start():
 
 
76
  files = None
77
 
78
  # Wait for the user to upload a file
@@ -142,14 +144,13 @@ async def main(message):
142
  print(debug_info)
143
 
144
  try:
145
- # If message is a string, wrap it in a cl.Message object
146
- if isinstance(message, str):
147
- message = cl.Message(content=message)
148
 
149
  chain = cl.user_session.get("chain") # type: ConversationalRetrievalChain
150
  cb = cl.AsyncLangchainCallbackHandler()
151
 
152
- res = await chain.acall(message.content, callbacks=[cb])
153
  answer = res["answer"]
154
  source_documents = res["source_documents"] # type: List[Document]
155
 
 
72
 
73
 
74
  @cl.on_chat_start
75
+ async def start():
76
+ await cl.Message("Welcome! Chat session started. You can now ask questions about the uploaded PDF.").send()
77
+
78
  files = None
79
 
80
  # Wait for the user to upload a file
 
144
  print(debug_info)
145
 
146
  try:
147
+ # If message is a string, use it directly
148
+ message_content = message if isinstance(message, str) else message.content
 
149
 
150
  chain = cl.user_session.get("chain") # type: ConversationalRetrievalChain
151
  cb = cl.AsyncLangchainCallbackHandler()
152
 
153
+ res = await chain.acall(message_content, callbacks=[cb])
154
  answer = res["answer"]
155
  source_documents = res["source_documents"] # type: List[Document]
156