Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import requests
 import json
+from PIL import Image
+from io import BytesIO
 
 def create_template(name, image_url, category, prompt):
     url = "https://ai-agent-backend-v50m.onrender.com/api/v1/templates/" 
@@ -39,6 +41,17 @@ def get_templates(skip=0, limit=10):
     except requests.RequestException as e:
         return f"Request failed: {str(e)}"
 
+def display_image_from_url(image_url):
+    try:
+        response = requests.get(image_url)
+        if response.status_code == 200:
+            image = Image.open(BytesIO(response.content))
+            st.image(image, caption="Template Image", use_column_width=True)
+        else:
+            st.error(f"Failed to load image from URL: {image_url}")
+    except Exception as e:
+        st.error(f"Error displaying image: {str(e)}")
+
 st.title("AI Agent Template Creator")
 
 # Create tabs for different functionalities
@@ -59,7 +72,7 @@ with tab1:
             st.error("Please fill in all fields including the image URL.")
 
     if image_url:
-        st.image(image_url, caption="Preview Image", use_column_width=True)
+        display_image_from_url(image_url)
 
 with tab2:
     st.header("Existing Templates")
@@ -79,7 +92,7 @@ with tab2:
                     st.write(f"Category: {template.get('category', 'N/A')}")
                     st.write(f"Prompt: {template.get('prompt', 'N/A')}")
                     if template.get('image'):
-                        st.image(template['image'], caption="Template Image", use_column_width=True)
+                        display_image_from_url(template['image'])
                     st.write("---")
         else:
             st.error(f"Failed to fetch templates: {templates}")