Save

app.py

@@ -102,23 +102,23 @@ def parse_response(response):
         print(f"Error parsing response: {e}")
         return []
 
-import re
-
 def cleanup_text(text):
     """
-    Strips braces, brackets, and commas from the given text.
+    Strips braces, brackets, commas, and double spaces from the given text.
 
     Args:
-    text: The text to parse.
+      text: The text to parse.
 
     Returns:
-    The parsed text.
+      The parsed text.
     """
 
     # Remove braces and brackets
     text = re.sub(r'[{}\[\]\'"]', '', text)
     # Remove commas
     text = text.replace(',', '')
+    # Remove double spaces
+    text = re.sub(' +', ' ', text)
     return text
 
 # Function to update session state when selection changes
@@ -159,7 +159,7 @@ def main():
     st.header("Step 1: Select a Framework")
 
     # Create a radio button
-    options = ["Atreamlit", "Gradio"]
+    options = ["Streamlit", "Gradio"]
     framework = st.radio(
         "Choose a framework:",
         options,
@@ -181,7 +181,8 @@ def main():
     framework = {st.session_state['selected_option']}
     selected_task = {st.session_state['selected_task']}
 
-    ai_prompt = f"Generate a {framework} app as follows: {selected_task}"
+    ai_prompt = f"Generate a {framework} app as follows: \
+        {selected_task}. Make sure the code is using the latest version of {framework}."
     ai_prompt = cleanup_text(ai_prompt)
 
     st.header("Step 2: Review and Customize the Prompt")