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!"})