image url fix

README.md

@@ -15,3 +15,6 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/644252f59a5bbb07ab7a987a/f4xH6ZNx1rVbKzr8jAHmt.png)
 
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/644252f59a5bbb07ab7a987a/A9BoWW2ulZycv0OQaihKO.png)
+
+
+[![Generated Graphic](https://api.slidegen.civai.co/images/image_20240606065804.png)](https://api.slidegen.civai.co/images/image_20240606065804.png)

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import requests, base64, io
 from dotenv import load_dotenv
-import os, datetime
+import os, datetime, mimetypes
 from PIL import Image
 load_dotenv()
 API_URL = os.environ.get("API_URL", "http://0.0.0.0:8021")
@@ -9,19 +9,31 @@ APP_ENV = os.environ.get("APP_ENV", None)
 
 
 def image_to_base64(input_file):
+    file_bytes = None
+    file_path = None
+
     if input_file is None:
         return None
 
-    file_bytes = None
     # Check if the input is a string (file path)
     if isinstance(input_file, str):
+
+        file_path = input_file
         with open(input_file, "rb") as file:
             file_bytes = file.read()
     else:
         # Handle file-like object (_TemporaryFileWrapper)
+        file_path = input_file.name
         file_bytes = input_file.read()
 
-    return base64.b64encode(file_bytes).decode()
+    # Guess the MIME type of the file based on the file extension
+    mime_type, _ = mimetypes.guess_type(file_path)
+    if mime_type is None:
+        mime_type = 'application/octet-stream'  # Use a binary data MIME type as a fallback
+
+    data_url = f"data:{mime_type};base64,{base64.b64encode(file_bytes).decode()}"
+
+    return data_url
 
 
 # Assuming the API returns an image URL in the response