Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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 |
-
|
| 125 |
-
|
|
|
|
|
|
|
| 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({
|