Update app.py

app.py

@@ -101,13 +101,17 @@ index_html = """
 
         <button type="submit">Generate Image</button>
     </form>
+    {% if image_url %}
+    <h2>Generated Image:</h2>
+    <img src="{{ image_url }}" alt="Generated Image" style="max-width: 100%; height: auto;">
+    {% endif %}
 </body>
 </html>
 """
 
 @app.route('/')
 def index():
-    return render_template_string(index_html)
+    return render_template_string(index_html, image_url=None)
 
 @app.route('/generate', methods=['GET'])
 def generate_image():
@@ -133,8 +137,17 @@ def generate_image():
     image.save(img_io, 'PNG')
     img_io.seek(0)
 
-    # Return the image
-    return send_file(img_io, mimetype='image/png')
+    # Create a unique filename for the image
+    image_filename = f"generated_image_{random.randint(1, 10000)}.png"
+    
+    # Save the image to a file
+    image.save(image_filename)
+
+    # Return the index page with the generated image URL
+    image_url = f"/static/{image_filename}"
+    return render_template_string(index_html, image_url=image_url)
 
 if __name__ == '__main__':
+    # Ensure the 'static' folder exists for saving images
+    os.makedirs('static', exist_ok=True)
     app.run(host='0.0.0.0', port=7860)