Create main.py
Browse files
main.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from fastapi import FastAPI, HTTPException,Query,Request,BackgroundTasks
|
| 2 |
+
import json
|
| 3 |
+
import os
|
| 4 |
+
import requests
|
| 5 |
+
from extract_insights import extract_fashion_categories
|
| 6 |
+
|
| 7 |
+
app = FastAPI()
|
| 8 |
+
|
| 9 |
+
@app.post("/process_ocr_text_openai_trigger")
|
| 10 |
+
async def process_document_base64(request: Request,background_tasks: BackgroundTasks):
|
| 11 |
+
# request_data = await request.json()
|
| 12 |
+
# user_id = request_data.get('user_id')
|
| 13 |
+
# message_id = request_data.get('message_id')
|
| 14 |
+
# email = request_data.get('email')
|
| 15 |
+
# await extract_fashion_categories(user_id,email,message_id)
|
| 16 |
+
# # background_tasks.add_task(extract_fashion_categories,user_id,email,message_id)
|
| 17 |
+
|
| 18 |
+
return {"message":'Endpoint working !'}
|