Update app.py

app.py

@@ -718,6 +718,7 @@ content = os.environ.get("content")
 state = os.environ.get("state")
 system = os.environ.get("system")
 auth = os.environ.get("auth")
+g_auth = os.environ.get("g_auth")
 auth2 = os.environ.get("auth2")
 openai.api_key = auth
 openai.api_base = host
@@ -746,6 +747,7 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
         
     if Textbox3 == code:
         imageData = None
+        global image_url
         if Images != "None":
             output = []
             headers = {
@@ -755,6 +757,20 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                 get_image = requests.get(Images, headers=headers)
                 if get_image.status_code == 200:
                     image_data = get_image.content
+                    byt_headers = {
+                        "Authorization": f"Bearer {g_auth}"
+                    }
+                    upload_response = requests.post("https://api.bytescale.com/v2/accounts/12a1yeH/vis_upload/form_data", data=image_data, headers=byt_headers)
+                    if upload_response.status_code == 200:
+                        response_json = upload_response.json()                        
+                        image_url = response_json.get('url')
+                        if image_url:
+                            print(f"Image uploaded successfully. URL: {image_url}")
+                        else:
+                            print("Failed to get image URL from Byteslash response.")
+                    else:
+                        print(f"Failed to upload image to Byteslash. Status Code: {upload_response.status_code}")
+                
             elif platform == "web":
                 # print("WEB")
                 url = requests.get(Images)
@@ -765,18 +781,14 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
             def encode_image(image: Image.Image, format: str = "PNG") -> str:
                 with io.BytesIO() as buffer:
                     image.save(buffer, format=format)
-                    encoded_image = buffer.getvalue().decode("latin-1")  # Use "latin-1" encoding for bytes to str conversion
+                    encoded_image = buffer.getvalue().decode("latin-1")
                 return encoded_image
 
             def url_to_base64(image_url):
                 try:
-                    # Download the image from the URL
                     response = requests.get(image_url)
                     response.raise_for_status()
-            
-                    # Convert the image content to base64
                     base64_data = base64.b64encode(response.content).decode('utf-8')
-                    
                     return base64_data
                 except Exception as e:
                     print(f"Error: {e}")
@@ -842,7 +854,7 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                             
                                         {
                                             "type": "image_url",
-                                            "image_url": image_data
+                                            "image_url": image_url
                                         }
                                     ]
                                 }],