|
from flask import Flask, jsonify |
|
import asyncio |
|
import os |
|
from hiou import hugging, Ahugging, browser |
|
import json |
|
|
|
async def get_answer(image: str, question: str): |
|
filename = image |
|
spid = os.getenv('SPID') |
|
hug = Ahugging(spid) |
|
filelink = await hug.upload(open(filename, "rb")) |
|
data = [ |
|
{ |
|
"meta": { |
|
"_type": "gradio.FileData" |
|
}, |
|
"mime_type": "image/jpeg", |
|
"orig_name": filename, |
|
"path": filelink.split("=", 1)[1], |
|
"size": os.path.getsize(filename), |
|
"url": filelink |
|
}, |
|
question, |
|
{ |
|
"tab_index": 0 |
|
} |
|
] |
|
datas = os.getenv('DATA') |
|
jsnd = json.loads(datas) |
|
data.append(jsnd) |
|
print(f"SPID : {spid}, DATA : {data}") |
|
await hug.filnal_setup(data, 2, 15) |
|
await hug.start() |
|
return hug.output.get("data")[0] |
|
|
|
app = Flask(__name__) |
|
|
|
|
|
@app.route('/async-task') |
|
async def async_task(): |
|
await asyncio.sleep(2) |
|
return jsonify({"message": "Async task completed!"}) |
|
|
|
@app.route('/') |
|
def home(): |
|
return jsonify({"message": "Welcome to my Flask API!"}) |
|
|
|
|