flaskapp / main.py
5m4ck3r's picture
Update main.py
26e1774 verified
raw
history blame
1.27 kB
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__)
# Example async route using asyncio.sleep
@app.route('/async-task')
async def async_task():
await asyncio.sleep(2) # Simulate async task
return jsonify({"message": "Async task completed!"})
@app.route('/')
def home():
return jsonify({"message": "Welcome to my Flask API!"})