37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
|
"""
|
|||
|
|
这是对于清华智谱VisualGLM-6B的代码，在使用前请先安装并启动好VisualGLM-6B.
|
|||
|
|
https://github.com/THUDM/VisualGLM-6B
|
|||
|
|
"""
|
|||
|
|
import json
|
|||
|
|
import requests
|
|||
|
|
import uuid
|
|||
|
|
import os
|
|||
|
|
import cv2
|
|||
|
|
from ai_module import yolov8
|
|||
|
|
|
|||
|
|
# Initialize an empty history list
|
|||
|
|
communication_history = []
|
|||
|
|
|
|||
|
|
def question(cont, uid=0, observation=""):
|
|||
|
|
if not yolov8.new_instance().get_status():
|
|||
|
|
return "请先启动“Fay Eyes”"
|
|||
|
|
content = {
|
|||
|
|
"text":cont,
|
|||
|
|
"history":communication_history}
|
|||
|
|
img = yolov8.new_instance().get_img()
|
|||
|
|
if yolov8.new_instance().get_status() and img is not None:
|
|||
|
|
filename = str(uuid.uuid4()) + ".jpg"
|
|||
|
|
current_working_directory = os.getcwd()
|
|||
|
|
filepath = os.path.join(current_working_directory, "data", filename)
|
|||
|
|
cv2.imwrite(filepath, img)
|
|||
|
|
content["image"] = filepath
|
|||
|
|
url = "http://127.0.0.1:8080"
|
|||
|
|
print(content)
|
|||
|
|
req = json.dumps(content)
|
|||
|
|
headers = {'content-type': 'application/json'}
|
|||
|
|
r = requests.post(url, headers=headers, data=req)
|
|||
|
|
|
|||
|
|
# Save this conversation to history
|
|||
|
|
communication_history.append([cont, r.text])
|
|||
|
|
|
|||
|
|
return r.text + "\n(相片:" + filepath + ")"
|