Update app.py

app.py

@@ -261,10 +261,7 @@ def get_user_input(config):
 
     custom_prompt_enabled = checkbox_with_default("Enable Custom Prompt", False, "custom_prompt_enabled")
 
-    if custom_prompt_enabled:
-        custom_prompt = st.text_area("Enter Custom Prompt", key="custom_prompt")
-    else:
-        custom_prompt = None
+    custom_prompt = st.text_area("Enter Custom Prompt", "Create a Lua script for Roblox:", key="custom_prompt")
 
     return {
         "color": color, "size": size, "position": position, "shape": shape, "material": material,
@@ -286,79 +283,8 @@ def get_user_input(config):
 
 def generate_lua_script(params, api_key):
     client = Groq(api_key=api_key)
-    prompt = params["custom_prompt"] if params["custom_prompt"] else "Create a Lua script for Roblox that:\n"
+    prompt = params["custom_prompt"]
     
-    if params["color"]:
-        prompt += f"1. Creates a {params['shape']} with the specified color {params['color']}.\n"
-    if params["size"]:
-        prompt += f"2. Specifies the size {params['size']}.\n"
-    if params["position"]:
-        prompt += f"3. Positions it at {params['position']}.\n"
-    if params["part_cframe"]:
-        prompt += f"4. Sets its CFrame to {params['part_cframe']}.\n"
-    if params["material"]:
-        prompt += f"5. Sets the material to {params['material']}.\n"
-    if params["transparency"]:
-        prompt += f"6. Sets the transparency to {params['transparency']}.\n"
-    if params["reflectance"]:
-        prompt += f"7. Sets the reflectance to {params['reflectance']}.\n"
-    if params["rotation_speed"]:
-        prompt += f"8. Rotates the {params['shape']} at a speed of {params['rotation_speed']}.\n"
-    if params["bounce_height"]:
-        prompt += f"9. Sets the bounce height to {params['bounce_height']}.\n"
-    if params["collision"]:
-        prompt += f"10. Sets collision to {params['collision']}.\n"
-    if params["surface"]:
-        prompt += f"11. Sets the surface type to {params['surface']}.\n"
-    if params["anchor"]:
-        prompt += f"12. Sets anchor to {params['anchor']}.\n"
-    if params["collision_group"]:
-        prompt += f"13. Sets the collision group to {params['collision_group']}.\n"
-    if params["angular_velocity"] and params["linear_velocity"]:
-        prompt += f"14. Sets the angular velocity to {params['angular_velocity']} and linear velocity to {params['linear_velocity']}.\n"
-    if params["density"]:
-        prompt += f"15. Sets the density to {params['density']}.\n"
-    if params["camera_type"]:
-        prompt += f"16. Sets the camera type to {params['camera_type']}.\n"
-    if params["camera_position"]:
-        prompt += f"17. Sets the camera position to {params['camera_position']}.\n"
-    if params["field_of_view"]:
-        prompt += f"18. Sets the field of view to {params['field_of_view']}.\n"
-    if params["camera_interpolation"]:
-        prompt += f"19. Enables camera interpolation.\n"
-    if params["ambient_light"]:
-        prompt += f"20. Sets the ambient light to {params['ambient_light']}.\n"
-    if params["brightness"]:
-        prompt += f"21. Sets the brightness to {params['brightness']}.\n"
-    if params["time_of_day"]:
-        prompt += f"22. Sets the time of day to {params['time_of_day']}.\n"
-    if params["shadow_softness"]:
-        prompt += f"23. Sets the shadow softness to {params['shadow_softness']}.\n"
-    if params["shadow_color"]:
-        prompt += f"24. Sets the shadow color to {params['shadow_color']}.\n"
-    if params["fog_start"] and params["fog_end"] and params["fog_color"]:
-        prompt += f"25. Sets fog with start distance {params['fog_start']}, end distance {params['fog_end']}, and color {params['fog_color']}.\n"
-    if params["walk_speed"]:
-        prompt += f"26. Sets the player's walk speed to {params['walk_speed']}.\n"
-    if params["jump_power"]:
-        prompt += f"27. Sets the player's jump power to {params['jump_power']}.\n"
-    if params["health"]:
-        prompt += f"28. Sets the player's health to {params['health']}.\n"
-    if params["respawn_time"]:
-        prompt += f"29. Sets the respawn time to {params['respawn_time']}.\n"
-    if params["background_music"]:
-        prompt += f"30. Adds background music with ID {params['background_music']} and sets volume to {params['sound_volume']}.\n"
-    if params["sound_effects"]:
-        prompt += f"31. Adds sound effects with ID {params['sound_effects']}.\n"
-    if params["environmental_sounds"]:
-        prompt += f"32. Adds environmental sounds with ID {params['environmental_sounds']}.\n"
-    if params["ui_text"]:
-        prompt += f"33. Adds UI text '{params['ui_text']}' at position {params['ui_position']} with size {params['ui_size']} and color {params['ui_color']} using font {params['ui_font']} and text size {params['ui_text_size']}.\n"
-    if params["texture_id"] and params["decal_position"]:
-        prompt += f"34. Adds a decal with texture ID {params['texture_id']} at position {params['decal_position']}.\n"
-    if params["particle_texture"]:
-        prompt += f"35. Adds particle effects with texture ID {params['particle_texture']}, lifetime {params['particle_lifetime']}, rate {params['particle_rate']}, speed {params['particle_speed']}, size {params['particle_size']}, acceleration {params['particle_acceleration']}, rotation speed {params['particle_rot_speed']}, and color {params['particle_color']}.\n"
-
     completion = client.chat.completions.create(
         model="llama3-8b-8192",
         messages=[
@@ -388,6 +314,7 @@ def main():
         api_key = config["api_key"]
         lua_script = generate_lua_script(user_input, api_key)
         st.code(lua_script, language='lua')
+        st.download_button(label="Copy to Clipboard", data=lua_script, file_name="lua_script.txt", mime="text/plain")
 
 if __name__ == "__main__":
     main()