NitinBot001 commited on
Commit
08bafde
·
verified ·
1 Parent(s): 723a9f5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -2
main.py CHANGED
@@ -121,8 +121,10 @@ def process_query():
121
  classification = SESSIONS[session_id].get('classification')
122
  category = classification['category']
123
  endpoint_url = API_ENDPOINTS.get(category)
124
- payload={"query": query}
125
- response = requests.post(endpoint_url,data=payload, json={"main_query": query})
 
 
126
  del SESSIONS[session_id]
127
  print(f"Session {session_id} closed.")
128
  return jsonify({
 
121
  classification = SESSIONS[session_id].get('classification')
122
  category = classification['category']
123
  endpoint_url = API_ENDPOINTS.get(category)
124
+ if category == "medicine_info":
125
+ response = requests.post(endpoint_url, json={"main_query": query})
126
+ else:
127
+ response = requests.post(endpoint_url, data={"main_query": query})
128
  del SESSIONS[session_id]
129
  print(f"Session {session_id} closed.")
130
  return jsonify({