Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import json
-import requests
+from groq import Groq
 
 def load_config():
     with open('config.json', 'r') as f:
@@ -285,94 +285,100 @@ def get_user_input(config):
     }
 
 def generate_lua_script(params, api_key):
-    url = "https://api.groq.com/v1/lua_script"  # Example URL, replace with actual API endpoint
-    headers = {
-        "Authorization": f"Bearer {api_key}",
-        "Content-Type": "application/json"
-    }
+    client = Groq(api_key=api_key)
+    prompt = params["custom_prompt"] if params["custom_prompt"] else "Create a Lua script for Roblox that:\n"
     
-    if params["custom_prompt"]:
-        prompt = params["custom_prompt"]
-    else:
-        prompt = "Create a Lua script for Roblox that:\n"
-        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"
-
-    data = {
-        "prompt": prompt
-    }
-
-    response = requests.post(url, headers=headers, json=data)
-    response.raise_for_status()  # Raise an exception for HTTP errors
-    return response.json()['script']
+    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=[
+            {
+                "role": "user",
+                "content": prompt
+            }
+        ],
+        temperature=1,
+        max_tokens=1024,
+        top_p=1,
+        stream=True,
+        stop=None,
+    )
+
+    lua_script = ""
+    for chunk in completion:
+        lua_script += chunk.choices[0].delta.content or ""
+
+    return lua_script
 
 def main():
     config = load_config()