Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+import requests
+import os
 
 
 
@@ -6,17 +8,39 @@ import gradio as gr
 import base64
 # Function to encode the image
 def encode_image(image_path):
-  with open(image_path, "rb") as image_file:
+    with open(image_path, "rb") as image_file:
     return base64.b64encode(image_file.read()).decode('utf-8')
 
-import requests
-import os
 
 openai_api_key = os.environ.get('openai_api_key')
 
+# 将图片上传到google cloud storage
+def upload_image_to_gcs_blob(image):
+
+    google_creds = os.environ.get("GOOGLE_APPLICATION_CREDENTIALS_JSON")
+
+    creds_json = json.loads(google_creds)
+    credentials = service_account.Credentials.from_service_account_info(creds_json)
+
+    # 现在您可以使用这些凭证对Google Cloud服务进行认证
+    storage_client = storage.Client(credentials=credentials, project=creds_json['project_id'])
+
+    bucket_name = os.environ.get('bucket_name')
+    bucket = storage_client.bucket(bucket_name)
+    
+    destination_blob_name = os.path.basename(image)
+    blob = bucket.blob(destination_blob_name)
+
+    blob.upload_from_filename(image)
+
+    public_url = blob.public_url
+    
+    return public_url
+
+
 def ask_image(text,image,api_token=openai_api_key):
-  base64_image = encode_image(image)
-  messages=[
+    public_url = upload_image_to_gcs_blob(image)
+    messages=[
       {
         "role": "user",
         "content": [
@@ -24,44 +48,44 @@ def ask_image(text,image,api_token=openai_api_key):
           {
             "type": "image_url",
             "image_url": {
-                "url":f"data:image/jpeg;base64,{base64_image}"
-  #               "url": "https://upload.wikimedia.org/wikipedia/commons/thumb/d/dd/Gfp-wisconsin-madison-the-nature-boardwalk.jpg/2560px-Gfp-wisconsin-madison-the-nature-boardwalk.jpg",
+                # "url":f"data:image/jpeg;base64,{base64_image}"
+                "url": public_url
             },
           },
         ],
       }
     ]
-
-  # 请求头部信息
-  headers = {
+    
+    # 请求头部信息
+    headers = {
       'Authorization': f'Bearer {api_token}'
-  }
-
-  # 请求体信息
-  data = {
+    }
+    
+    # 请求体信息
+    data = {
       'model': 'gpt-4o',  # 可以根据需要更换其他模型
       'messages': messages,
       'temperature': 0.7  # 可以根据需要调整
-  }
-
-
-  # 设定最大重试次数
-  max_retry = 3
-
-  for i in range(max_retry):
+    }
+    
+    
+    # 设定最大重试次数
+    max_retry = 3
+    
+    for i in range(max_retry):
       try:
           # 发送请求
           response = requests.post('https://burn.hair/v1/chat/completions', headers=headers, json=data)
-
+    
           # 解析响应内容
           response_data = response.json()
           response_content = response_data['choices'][0]['message']['content']
           usage = response_data['usage']
-
+    
           # response_content = 'test response'
-
+    
           return response_content
-
+    
       except Exception as e:
           # 如果已经达到最大重试次数，那么返回空值
           if i == max_retry - 1: