vineelagampa AyushSankar13 commited on
Commit
79846f1
·
verified ·
1 Parent(s): 4b899e5

Update backend.py (#32)

Browse files

- Update backend.py (2c9f79409a1de4af643df37263d101bbe901994f)


Co-authored-by: Ayush Sankar <AyushSankar13@users.noreply.huggingface.co>

Files changed (1) hide show
  1. backend.py +5 -0
backend.py CHANGED
@@ -195,6 +195,11 @@ async def chat_endpoint(request: ChatRequest):
195
  document_text=document_text,
196
  user_question=request.question
197
  )
 
 
 
 
 
198
  response = model.generate_content(full_prompt)
199
  return ChatResponse(answer=response.text)
200
  except Exception as e:
 
195
  document_text=document_text,
196
  user_question=request.question
197
  )
198
+ data = await request.json()
199
+ question = data.get("question")
200
+ report_text = data.get("report_text")
201
+ user_id = data.get("user_id")
202
+
203
  response = model.generate_content(full_prompt)
204
  return ChatResponse(answer=response.text)
205
  except Exception as e: