Update app.py

app.py

@@ -483,7 +483,6 @@
 # )
 # iface.launch()
 
-
 import gradio as gr
 import numpy as np
 import cv2 as cv
@@ -516,6 +515,7 @@ messages = [{"role": "system", "content": system}]
 def classify(platform, UserInput, Images, Textbox2, Textbox3):
     if Textbox3 == code:
         imageData = None
+        image_data_url = None  # Initialize image_data_url
         if Images is not None:
             output = []
             headers = {
@@ -524,13 +524,12 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
             if platform == "wh":
                 get_image = requests.get(Images, headers=headers)
                 if get_image.status_code == 200:
-                    # Convert the image content to a base64 data URL
+                    # Correct base64 encoding and URL creation
                     image_base64 = base64.b64encode(get_image.content).decode("utf-8")
                     image_data_url = f"data:image/png;base64,{image_base64}"
 
                     random_id = random.randint(1000, 9999)
-                    file_extension = ".png"
-                    filename = f"image_{random_id}{file_extension}"
+                    filename = f"image_{random_id}"
                     with open(filename, "wb") as file:
                         file.write(get_image.content)
                     print(f"Saved image as: {filename}")
@@ -544,11 +543,11 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
             else:
                 pass
 
-            if imageData is not None:
-                image = cv.imread(full_path)
+            if image_data_url is not None:
+                # Load the image from image_data_url
+                image = cv.imread(image_data_url)
                 image = cv.resize(image, (224, 224))
                 image_array = np.asarray(image)
-                image_data = cv.resize(imageData, (224, 224))
                 normalized_image_array = (image_array.astype(np.float32) / 127.0) - 1
                 data[0] = normalized_image_array
 
@@ -620,7 +619,7 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
             messages.append({"role": "user", "content": UserInput})
 
             headers = {
-                "Content-Type": "application.json",
+                "Content-Type": "application/json",
                 "Authorization": f"Bearer {auth}"
             }