git ignore updated

.gitignore

@@ -0,0 +1,23 @@
+# Python cache files
+*.pyc
+__pycache__/
+
+# Temporary files
+temp_image.jpg
+*.tmp
+
+# Streamlit app files
+*.streamlit
+
+# Environment files
+.env
+
+# Logs
+*.log
+
+# OS-specific files
+.DS_Store
+Thumbs.db
+
+# Generated HTML files
+index.html

app.py

@@ -61,6 +61,13 @@ def main():
 
     if uploaded_file is not None:
         try:
+            if uploaded_file.size > 5 * 1024 * 1024:  # Limit file size to 5MB
+                st.error("File size exceeds the limit of 5MB.")
+                return
+            if uploaded_file.type not in ["image/jpeg", "image/png"]:
+                st.error("Invalid file type. Please upload a JPEG or PNG image.")
+                return
+
             # Load and display the image
             image = Image.open(uploaded_file)
             st.image(image, caption='Uploaded Image.', use_container_width=True)
@@ -105,7 +112,7 @@ def main():
                     st.write(f"You selected: {uiFrameWork}")
                 
                     generated_html = send_message_to_model(detailed_prompt, temp_image_path)
-                    st.code(generated_html, language='html')
+                    
 
                     # Save the refined HTML to a file
                     with open("index.html", "w") as file: