Update app.py (#1)

- Update app.py (cf714c5126d7d6975fc8bb3f8d615099b4a20a97)


Co-authored-by: Jim Pang <jim33282007@users.noreply.huggingface.co>

app.py

@@ -20,7 +20,7 @@ def load_image_from_url(url):
         return None
 
 # User option to select input type: Upload or URL
-input_type = st.radio("Select input type:", ("Upload Image", "Image URL"))
+input_type = st.radio("Select input type:", ("Upload Image", "Image URL", "Text"))
 
 if input_type == "Upload Image":
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
@@ -33,11 +33,19 @@ elif input_type == "Image URL":
         image = load_image_from_url(image_url)
         if image:
             st.image(image, caption='Image from URL', use_column_width=True)
+elif input_type == "Text":
+    text_input = st.text_input("Enter text here:", "")
+    if text_input:
+        st.image(image, caption='Image from URL', use_column_width=True)
 
 # Generate caption button
 if st.button('Generate Caption'):
     if not image:
-        st.warning("Please upload an image or enter an image URL.")
+        if not text_input:
+            st.warning("Please upload an image, enter an image URL or input text")
+        else:
+            result = text_input
+            
     else:
         with st.spinner("Generating caption..."):
             # Process the image and generate caption
@@ -49,8 +57,8 @@ if st.button('Generate Caption'):
             elif input_type == "Image URL" and image_url:
                 result = image_to_text(image_url)
                 
-            if result:
-                generated_text = result[0]['generated_text']
-                st.success(f'Generated Caption: {generated_text}')
-            else:
-                st.error("Failed to generate caption.")
+    if result:
+        generated_text = result[0]['generated_text']
+        st.success(f'Generated Caption: {generated_text}')
+    else:
+        st.error("Failed to generate caption.")