Update app.py

app.py

@@ -10,9 +10,6 @@ def send_text2image_request(prompt, ratio, style, uuid_value, token):
     if not url:
         return "Error: API URL not set in environment secrets."
 
-    # Print the URL for debugging purposes
-    
-    
     # Define the headers with the required values
     headers = {
         "Uid": uuid_value,  # Using the provided UUID
@@ -47,9 +44,6 @@ def send_image2image_request(image, ratio, style, uuid_value, token):
     if not url:
         return "Error: API URL not set in environment secrets."
 
-    # Print the URL for debugging purposes
-    
-    
     # Define the headers with the required values
     headers = {
         "Uid": uuid_value,  # Using the provided UUID
@@ -72,8 +66,7 @@ def send_image2image_request(image, ratio, style, uuid_value, token):
     if style and style != "0":
         data["style_id"] = style
 
-
-    response = requests.post(url, headers=headers, data=data, files=files)
+    response = requests.post(url, headers=headers, data=data)
 
     # Return the response text or any relevant information
     if response.status_code == 200:
@@ -81,7 +74,27 @@ def send_image2image_request(image, ratio, style, uuid_value, token):
     else:
         return f"Error: {response.status_code} - {response.text}"
 
-# Gradio interface with tabs for Text-to-Image and Image-to-Image
+# Function to get the Upload URL from Firebase
+def get_upload_url(uuid_value):
+    # The URL to send the request to
+    firebase_url = 'https://firebasestorage.googleapis.com/v0/b/hardstone_img_us/o?name=umagic_crop%2Fupload%2F2024-11-17%2F{}.jpg&uploadType=resumable'.format(uuid_value)
+
+    # Define the headers with the required values
+    headers = {
+        "Authorization": "Bearer YOUR_ACCESS_TOKEN",  # Replace with the correct Bearer Token if needed
+        "Content-Type": "application/json"
+    }
+
+    # Sending GET request to Firebase Storage
+    response = requests.get(firebase_url, headers=headers)
+
+    # Return the response text or any relevant information
+    if response.status_code == 200:
+        return response.json()  # You can return JSON data if that's the expected response format
+    else:
+        return f"Error: {response.status_code} - {response.text}"
+
+# Gradio interface with tabs for Text-to-Image, Image-to-Image, and Get Upload URL
 def create_interface():
     with gr.Blocks() as demo:
         with gr.Tab("Text to Image"):
@@ -109,7 +122,16 @@ def create_interface():
             # Bind the image-to-image function to the interface
             img_submit_button = gr.Button("Generate Image")
             img_submit_button.click(send_image2image_request, inputs=[img_url, img_ratio, img_style, img_uuid, img_token], outputs=img_output)
-    
+        
+        with gr.Tab("Get Upload URL"):
+            # Inputs for Get Upload URL
+            uuid_input = gr.Textbox(label="UUID ID")  # UUID for getting upload URL
+            upload_url_output = gr.Textbox()  # Output for the upload URL response
+            
+            # Bind the get_upload_url function to the interface
+            upload_submit_button = gr.Button("Get Upload URL")
+            upload_submit_button.click(get_upload_url, inputs=[uuid_input], outputs=upload_url_output)
+
     return demo
 
 # Launch the Gradio interface