add of variables

app.py

@@ -5,8 +5,16 @@ import requests
 import streamlit as st
 
 
-def encode_image(uploaded_file):
-    bytes_data = uploaded_file.getvalue()
+def encode_image(uploaded_file=None, image_url=None):
+    if uploaded_file:
+        # Handle file upload
+        bytes_data = uploaded_file.getvalue()
+    elif image_url:
+        # Handle image URL
+        response = requests.get(image_url)
+        bytes_data = response.content
+    else:
+        raise ValueError("Either uploaded_file or image_url must be provided.")
     encoded = base64.b64encode(bytes_data).decode()
     return f"data:image/jpeg;base64,{encoded}"
 
@@ -48,18 +56,45 @@ def main():
     token = st.sidebar.text_input("Token")
 
     # File uploader
-    uploaded_file = st.file_uploader(
+    col1, col2 = st.columns(2)
+
+    with col1:
+        uploaded_file = st.file_uploader(
         "Choose an image", type=["png", "jpg", "jpeg"]
     )
-
+    with col2:
+        image_url = st.text_input("Or enter an image URL", help="Remove the uploaded image if you want to use an image URL")
     if uploaded_file:
         st.image(uploaded_file, caption="Uploaded Image")
+    elif image_url:
+        st.image(image_url, caption="Image URL")
+    # Text inputs
+    input_category = st.text_input("Enter category (Optional)")
+    input_variant = st.text_input("Enter variant")
+
+    # Method input
+    option_labels = [
+    "Value 1: For replacement of element in foreground",
+    "Value 2: For replacement of background",
+    "Value 3: To generate a new image with the same dimensions based on the prompt",
+    "Value 4: Workflow stop",
+    ]
+
+    option_values = [1, 2, 3, 4]
+
+    selected_option = st.selectbox(
+        "Select an option:",
+        index=None,
+        options=option_labels,
+    )
+    if selected_option is None:
+        input_number = None
+    else:
+        input_number = option_values[option_labels.index(selected_option)]
 
-    # Text input
-    input_text = st.text_input("Enter text")
-
-    if st.button("Generate") and uploaded_file and input_text:
-        encoded_image = encode_image(uploaded_file)
+    # Generation
+    if st.button("Generate") and (uploaded_file or image_url) and input_variant:
+        encoded_image = encode_image(uploaded_file=uploaded_file, image_url=image_url)
 
         # Initial API call
         headers = {
@@ -69,11 +104,16 @@ def main():
 
         data = {
             "execution_mode": "async",
-            "inputs": {"input_image": encoded_image, "input_text": input_text},
+            "inputs": {"input_image": encoded_image, "input_text": input_variant},
             "workflow_version_id": "c5249acc-2cda-4734-b4f8-7823cecbce3d",
             "machine_id": "353b27c7-6bce-4472-b5eb-18f22d2373fc",
         }
 
+        if input_category:
+            data["inputs"]["input_category"] = input_category
+        if input_number:
+            data["inputs"]["input_number"] = input_number
+
         with st.spinner("Generating image..."):
             try:
                 response = requests.post(
@@ -107,6 +147,7 @@ def main():
 
                     counter += 1
                 outputs = [Output(x) for x in result["outputs"]]
+
                 final_img = [
                     output.output_url
                     for output in outputs
@@ -114,6 +155,7 @@ def main():
                 ][0]
                 st.image(final_img, caption="Generated Image")
 
+
             except requests.exceptions.RequestException as e:
                 st.error(f"Error occurred: {str(e)}")