5m4ck3r commited on
Commit
0d1fedf
·
verified ·
1 Parent(s): 635344d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +13 -10
main.py CHANGED
@@ -1,15 +1,14 @@
1
  from flask import Flask, request, jsonify
2
- import asyncio
3
  import os
4
- from hiou import hugging, browser
5
  import json
6
  import io
7
 
8
- def get_answer(image: str, question: str):
9
- filename = image
10
  spid = os.getenv('SPID')
11
  hug = hugging(spid)
12
- filelink = hug.upload(filename)
13
  image.seek(0, 2)
14
  file_size = image.tell()
15
  image.seek(0)
@@ -23,16 +22,19 @@ def get_answer(image: str, question: str):
23
  "path": filelink.split("=", 1)[1],
24
  "size": file_size,
25
  "url": filelink
26
- },
27
- question,
28
- {
29
- "tab_index": 0
30
  }
31
  ]
32
  datas = os.getenv('DATA')
33
  jsnd = json.loads(datas)
34
  data.append(jsnd)
35
- print(f"SPID : {spid}, DATA : {data}")
 
 
 
 
 
 
 
36
  hug.filnal_setup(data, 2, 15)
37
  hug.start()
38
  return hug.output.get("data")[0]
@@ -59,3 +61,4 @@ def async_task():
59
  def home():
60
  return jsonify({"message": "Welcome to my Flask API!"})
61
 
 
 
1
  from flask import Flask, request, jsonify
 
2
  import os
3
+ from hiou import hugging
4
  import json
5
  import io
6
 
7
+ def get_answer(image: io.BytesIO, question: str):
8
+ filename = image.name
9
  spid = os.getenv('SPID')
10
  hug = hugging(spid)
11
+ filelink = hug.upload(image)
12
  image.seek(0, 2)
13
  file_size = image.tell()
14
  image.seek(0)
 
22
  "path": filelink.split("=", 1)[1],
23
  "size": file_size,
24
  "url": filelink
 
 
 
 
25
  }
26
  ]
27
  datas = os.getenv('DATA')
28
  jsnd = json.loads(datas)
29
  data.append(jsnd)
30
+ data.append(
31
+ question
32
+ )
33
+ data.append(
34
+ {
35
+ "tab_index": 0
36
+ }
37
+ )
38
  hug.filnal_setup(data, 2, 15)
39
  hug.start()
40
  return hug.output.get("data")[0]
 
61
  def home():
62
  return jsonify({"message": "Welcome to my Flask API!"})
63
 
64
+ app.run()