khalednabawi11 commited on
Commit
29f410d
·
verified ·
1 Parent(s): 495be45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py CHANGED
@@ -191,6 +191,20 @@ async def ask(query: Query):
191
  detail={"status": "error", "message": "Internal server error", "error": str(e)}
192
  )
193
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
194
  # === ENTRYPOINT === #
195
  if __name__ == "__main__":
196
  def handle_exit(signum, frame):
 
191
  detail={"status": "error", "message": "Internal server error", "error": str(e)}
192
  )
193
 
194
+ @app.post("/chat")
195
+ def chat(query: Query):
196
+
197
+ prompt = generate_prompt(query.question)
198
+
199
+ answer = qa_chain.run(prompt)
200
+
201
+ return {
202
+
203
+ "answer": answer
204
+ }
205
+
206
+
207
+
208
  # === ENTRYPOINT === #
209
  if __name__ == "__main__":
210
  def handle_exit(signum, frame):