Update main.py

main.py

@@ -1,5 +1,37 @@
 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__)