Update app.py

app.py

@@ -7,7 +7,7 @@ import json
 from rembg import remove
 import requests
 from pathlib import Path
-import zipfile
+from streamlit_drawable_canvas import st_canvas
 
 # ------------------------------
 # Utility Functions
@@ -125,6 +125,12 @@ if 'history' not in st.session_state:
 if 'redo_stack' not in st.session_state:
     st.session_state['redo_stack'] = []
 
+# Initialize session state for text addition
+if 'text_input' not in st.session_state:
+    st.session_state['text_input'] = ""
+if 'click_position' not in st.session_state:
+    st.session_state['click_position'] = None
+
 # ------------------------------
 # Layer Management Functions
 # ------------------------------
@@ -462,14 +468,72 @@ if load_project:
 
 if uploaded_file:
     base_image = Image.open(uploaded_file).convert("RGBA")
-    canvas = base_image.copy()
+    canvas_image = base_image.copy()
 
     # Apply auto enhance if selected
     if apply_auto_enhance:
-        canvas = auto_enhance(canvas)
+        canvas_image = auto_enhance(canvas_image)
 
     # Initialize drawing context
-    draw = ImageDraw.Draw(canvas, 'RGBA')
+    draw = ImageDraw.Draw(canvas_image, 'RGBA')
+
+    # ------------------------------
+    # Interactive Text Addition
+    # ------------------------------
+
+    st.header("🖌️ Interactive Text Addition")
+
+    # Display the canvas with drawable features
+    canvas_result = st_canvas(
+        fill_color="rgba(255, 255, 255, 0)",  # Transparent fill
+        stroke_width=0,
+        stroke_color="#000000",
+        background_image=canvas_image,
+        update_streamlit=True,
+        height=canvas_image.height,
+        width=canvas_image.width,
+        drawing_mode="freedraw",
+        key="canvas",
+        # Hide the drawing toolbar
+        display_toolbar=False,
+    )
+
+    if canvas_result.json_data is not None:
+        objects = canvas_result.json_data["objects"]
+        if objects:
+            last_object = objects[-1]
+            if last_object["type"] == "point":
+                x, y = last_object["left"], last_object["top"]
+                st.session_state['click_position'] = (int(x), int(y))
+                st.session_state['text_input'] = st.text_input("Enter Text for the Clicked Position:", "")
+                if st.button("Add Text Here"):
+                    text_content = st.session_state['text_input']
+                    if text_content:
+                        # Define default properties for interactive text
+                        layer = {
+                            'type': 'Text',
+                            'content': text_content,
+                            'x': st.session_state['click_position'][0],
+                            'y': st.session_state['click_position'][1],
+                            'font_size': 40,
+                            'font_color': "#000000",
+                            'font_path': list(font_paths.values())[0],  # Default to the first font
+                            'opacity': 100,
+                            'visible': True,
+                            'alignment': "left",
+                            'blending_mode': "Normal"
+                        }
+                        add_layer(layer)
+                        st.success("Interactive text added!")
+                        # Reset click position and text input
+                        st.session_state['click_position'] = None
+                        st.session_state['text_input'] = ""
+                    else:
+                        st.warning("Please enter some text.")
+    
+    # ------------------------------
+    # Process Layers and Render Image
+    # ------------------------------
 
     for layer in st.session_state['layers']:
         if not layer.get('visible', True):
@@ -487,15 +551,15 @@ if uploaded_file:
             except ValueError:
                 text_color_rgba = (0, 0, 0, text_opacity)
             text_color_rgba = text_color_rgba[:3] + (text_opacity,)
-            text_layer = Image.new('RGBA', canvas.size, (255,255,255,0))
+            text_layer = Image.new('RGBA', canvas_image.size, (255,255,255,0))
             text_draw = ImageDraw.Draw(text_layer)
             alignment = layer.get('alignment', 'left')
             text_position = (layer['x'], layer['y'])
             text_draw.text(text_position, layer['content'], fill=text_color_rgba, font=font, align=alignment)
             if layer.get('blending_mode') and layer['blending_mode'] != "Normal":
-                canvas = blend_images(canvas, text_layer, layer['blending_mode'])
+                canvas_image = blend_images(canvas_image, text_layer, layer['blending_mode'])
             else:
-                canvas = Image.alpha_composite(canvas, text_layer)
+                canvas_image = Image.alpha_composite(canvas_image, text_layer)
         elif layer['type'] in ["Image", "Sticker"]:
             overlay = layer['content'].copy()
             opacity = layer.get('opacity', 100)
@@ -504,11 +568,11 @@ if uploaded_file:
                 alpha = alpha.point(lambda p: p * opacity / 100)
                 overlay.putalpha(alpha)
             if layer.get('blending_mode') and layer['blending_mode'] != "Normal":
-                canvas = blend_images(canvas, overlay, layer['blending_mode'])
+                canvas_image = blend_images(canvas_image, overlay, layer['blending_mode'])
             else:
-                canvas.paste(overlay, (layer['x'], layer['y']), overlay)
+                canvas_image.paste(overlay, (layer['x'], layer['y']), overlay)
         elif layer['type'] == "Shape":
-            shape_layer = Image.new('RGBA', canvas.size, (255,255,255,0))
+            shape_layer = Image.new('RGBA', canvas_image.size, (255,255,255,0))
             shape_draw = ImageDraw.Draw(shape_layer)
             color = layer['color']
             opacity = layer.get('opacity', 100)
@@ -526,27 +590,27 @@ if uploaded_file:
             elif layer['shape'] == "Line":
                 shape_draw.line([x1, y1, x2, y2], fill=shape_color_rgba, width=thickness)
             if layer.get('blending_mode') and layer['blending_mode'] != "Normal":
-                canvas = blend_images(canvas, shape_layer, layer['blending_mode'])
+                canvas_image = blend_images(canvas_image, shape_layer, layer['blending_mode'])
             else:
-                canvas = Image.alpha_composite(canvas, shape_layer)
+                canvas_image = Image.alpha_composite(canvas_image, shape_layer)
 
     # Apply filters
-    canvas = apply_filters(canvas, filters)
+    canvas_image = apply_filters(canvas_image, filters)
 
     # Remove background if selected
     if remove_bg:
         try:
-            canvas = remove_background(canvas)
+            canvas_image = remove_background(canvas_image)
             st.success("Background removed successfully!")
         except Exception as e:
             st.error(f"Background removal failed: {e}")
 
     # Display the final image
-    st.image(canvas, caption="Edited Image", use_column_width=True)
+    st.image(canvas_image, caption="Edited Image", use_column_width=True)
 
     # Download options
     buffer = io.BytesIO()
-    canvas.save(buffer, format="PNG")
+    canvas_image.save(buffer, format="PNG")
     st.download_button(
         label="📥 Download Edited Image",
         data=buffer.getvalue(),