Update app.py

app.py

@@ -47,28 +47,29 @@ def main():
 
     with right:
         if uploaded_image:
-            with st.spinner("Generating HTML + CSS using Gemini 2.5 Flash..."):
-                try:
-                    output = generate_html_css_from_image(uploaded_image)
-                    if output:
-                        st.subheader("💬 Generated HTML + CSS:")
-                        placeholder = st.empty()
-                        typed_text = ""
-                        for char in output:
-                            typed_text += char
-                            placeholder.code(typed_text, language="html")
-                            time.sleep(0.003)
+            if "output_html" not in st.session_state or st.session_state.get("last_image") != uploaded_image.name:
+                with st.spinner("Generating HTML + CSS using Gemini 2.5 Flash..."):
+                    try:
+                        output = generate_html_css_from_image(uploaded_image)
+                        st.session_state.output_html = output
+                        st.session_state.last_image = uploaded_image.name
+                    except Exception as e:
+                        st.error(f"Error: {e}")
+                        return
 
-                        st.download_button(
-                            label="💾 Download HTML file",
-                            data=output,
-                            file_name="generated_ui.html",
-                            mime="text/html"
-                        )
-                    else:
-                        st.error("No output generated. Please try again.")
-                except Exception as e:
-                    st.error(f"Error: {e}")
+        output = st.session_state.output_html
+        if output:
+            st.subheader("💬 Generated HTML + CSS:")
+            placeholder = st.empty()
+            typed_text = ""
+            for char in output:
+                typed_text += char
+                placeholder.code(typed_text, language="html")
+                time.sleep(0.002)
+
+            st.download_button("💾 Download HTML file", data=output, file_name="generated_ui.html", mime="text/html")
+        else:
+            st.error("No output generated.")
 
 if __name__ == "__main__":
     main()