switch to md

app.py

@@ -8,60 +8,19 @@ API_URL = os.environ.get("API_URL", "http://0.0.0.0:8021")
 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
-
-    # 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()
-
-    # 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
-def generate_image(brand_name, primary_color, secondary_color, description, target_audience, font, logo_image, heading_text, sub_heading_text, image_layout, custom_graphic_prompt):
-
-    if logo_image:
-        logo_base64 = image_to_base64(logo_image)
-    else:
-        logo_base64 = ""
+def generate_image(campaign_details):
 
 
     # Define your payload/data to send to the image generation API
     data = {
-        "brand_name": brand_name,
-        "primary_color": primary_color,
-        "secondary_color": secondary_color,
-        "description": description,
-        "target_audience": target_audience,
-        "font": font,
-        "heading_text": heading_text or "",
-        "sub_heading_text": sub_heading_text or "",
-        "logo_url": logo_base64,
-        "user_prompt": custom_graphic_prompt,
-        "image_dimension": image_layout
+        "request_string": campaign_details,
+        "is_single_image": True
     }
 
     # Make the API call
-    response = requests.post(API_URL, json=data)
+    response = requests.post(API_URL + "/generate_graphic", json=data)
 
     # Ensure the API call was successful
     if response.status_code != 200:
@@ -98,29 +57,59 @@ content_html = """
 
 """
 
+default_value = """
+# Brand Name
+Enter your brand name here
+
+## Description
+Enter your description here.
+
+---
+
+## Branding
+
+### Colors
+Primary Color: i.e #000000
+Secondary Color: i.e #FFFFFF
+
+### Font
+Enter font here i.e Arial
+
+### Logo
+Enter your logo url here.
+
+---
+
+## Audience
+Enter your target audience here.
+
+---
+
+## Headings
+### Main Heading
+Your main heading text here.
+
+### Sub-heading
+Your sub-heading text here.
+
+---
+
+## Image Layout
+### Select Layout
+Layout can be any of: Mobile Portrait (750x1334), Mobile Landscape (940x470), Square (2048x2048)
+
+Selected Layout: Mobile Portrait (750x1334)
+
+---
+
+## Custom Graphics
+Enter your custom graphic prompt here.
+"""
+
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.components.Textbox(placeholder="Brand/Company Name", label="Brand Name"),
-        gr.components.ColorPicker(value="#FFFFFF", label="Primary Color"),
-        gr.components.ColorPicker(value="#05A592", label="Secondary Color"),
-        gr.components.Textbox(placeholder="Brief description of your brand and design guidelines", label="Description", lines=2),
-        gr.components.Textbox(placeholder="Your Brand's target audience", label="Target Audience"),
-        gr.components.Textbox(placeholder="Google Font Label i.e Rubik", label="Font"),
-        gr.components.File(label="Logo Image", scale=2, file_types=["jpg", "jpeg", "png", "gif", "bmp"]),
-        gr.components.Textbox(placeholder="Heading text (not required)", label="Heading Text"),
-        gr.components.Textbox(placeholder="Paragraph text (not required)", label="Sub Heading Text"),
-        gr.components.Dropdown(
-            choices=[
-                ("Select Layout", "750x1334"),
-                ("Mobile Portrait (750x1334)", "750x1334"),
-                ("Mobile Landscape (940x470)", "940x470"),
-                ("Square (2048x2048)", "2048x2048")
-            ],
-            value="750x1334",
-            label="Image Layout"
-        ),
-        gr.components.Textbox(lines=4, placeholder="Enter your custom dall-e 2-3 image prompt here (not required)", value="", label="Custom Graphic Prompt")
+        gr.components.Textbox(lines=10, placeholder="Enter your Brand and campaign details here", value=default_value, label="Campaign Details")
     ],
     outputs=[
         gr.components.Image(label="Generated Image")