Update app.py

app.py

@@ -33,3 +33,57 @@ def generate_app(description, bot):
 def refine_app(current_code, refinement, bot):
     chat = ChatOpenAI(temperature=0.1, model_name="gpt-4", openai_api_key=OPENAI_API_KEY)
     prompt = f"""@{bot} I have a Poe Canvas App that I want to refine. Here is the current code:
+
+I want to make the following changes:
+"{refinement}"
+
+Please provide an updated version of the code that implements these changes. Keep the core functionality and only modify what's necessary.
+
+Your response should follow this format:
+
+1. A brief explanation of what changes you've made
+2. A Markdown code block containing the complete updated HTML/CSS/JS for the app"""
+    return chat.chat(messages_from_template(prompt))
+
+def main():
+    st.title("App Creator AI")
+    st.header("Create and refine your web applications")
+
+    # App Description Input
+    with st.expander("1. App Description"):
+        description = st.text_area(
+            "Describe the app you want to create",
+            placeholder="Describe the app you want to create in detail. For example: 'Create a to-do list app with the ability to add, remove, and mark tasks as complete. The app should have a clean design with a dark mode option.'"
+        )
+        bot = st.selectbox(
+            "AI Model",
+            ["Claude-3.7-Sonnet", "GPT-4o", "GPT-4o-mini", "o3-mini"]
+        )
+        if st.button("Generate App"):
+            if not description:
+                st.error("Please enter a description of the app you want to create.")
+                return
+            result = generate_app(description, bot)
+            # Display result in a code block
+            st.markdown("```html\n" + result + "\n```")
+
+    # Results Section
+    with st.expander("2. Generated App"):
+        code = st.text_area("Generated Code")
+        if code:
+            # Show preview in a frame
+            st.components.html(code, height=500)
+            # Add copy button
+            if st.button("Copy Code"):
+                st.write(code)
+    
+    # Refinement Section
+    with st.expander("3. Refine Your App"):
+        refinement = st.text_area("Enter refinements")
+        if refinement:
+            if st.button("Refine App"):
+                result = refine_app(code, refinement, bot)
+                st.markdown("```html\n" + result + "\n```")
+
+if __name__ == "__main__":
+    main()