Create main.py

main.py

@@ -0,0 +1,379 @@
+import streamlit as st
+import json
+from groq import Groq
+
+def load_config():
+    with open('config.json', 'r') as f:
+        return json.load(f)
+
+def get_user_input(config):
+    st.title("Advanced Roblox Lua Script Generator")
+
+    def checkbox_with_default(label, value):
+        return st.checkbox(label, value=value)
+
+    def text_input_with_default(label, value):
+        return st.text_input(label, value)
+
+    def selectbox_with_default(label, options, value):
+        return st.selectbox(label, options, index=options.index(value))
+
+    default_params = config["default_parameters"]
+
+    # Game Object Properties
+    color_enabled = checkbox_with_default("Enable Color", True)
+    size_enabled = checkbox_with_default("Enable Size", True)
+    position_enabled = checkbox_with_default("Enable Position", True)
+    shape_enabled = checkbox_with_default("Enable Shape", True)
+    material_enabled = checkbox_with_default("Enable Material", True)
+    transparency_enabled = checkbox_with_default("Enable Transparency", True)
+    reflectance_enabled = checkbox_with_default("Enable Reflectance", True)
+    rotation_speed_enabled = checkbox_with_default("Enable Rotation Speed", True)
+    bounce_height_enabled = checkbox_with_default("Enable Bounce Height", True)
+    collision_enabled = checkbox_with_default("Enable Collision", True)
+    surface_enabled = checkbox_with_default("Enable Surface Type", True)
+    anchor_enabled = checkbox_with_default("Enable Anchor", True)
+    collision_group_enabled = checkbox_with_default("Enable Collision Group", True)
+    custom_physics_enabled = checkbox_with_default("Enable Custom Physics", True)
+    part_cframe_enabled = checkbox_with_default("Enable Part CFrame", True)
+
+    # Camera Properties
+    camera_enabled = checkbox_with_default("Enable Camera Configuration", True)
+    camera_interpolation_enabled = checkbox_with_default("Enable Camera Interpolation", True)
+
+    # Lighting Properties
+    lighting_enabled = checkbox_with_default("Enable Lighting Configuration", True)
+    fog_enabled = checkbox_with_default("Enable Environmental Fog", True)
+    shadow_enabled = checkbox_with_default("Enable Shadow Properties", True)
+
+    # Player Properties
+    player_settings_enabled = checkbox_with_default("Enable Player Settings", True)
+    respawn_time_enabled = checkbox_with_default("Enable Respawn Time", True)
+
+    # Sound Properties
+    sound_enabled = checkbox_with_default("Enable Sound", True)
+    sound_effects_enabled = checkbox_with_default("Enable Sound Effects", True)
+    environmental_sounds_enabled = checkbox_with_default("Enable Environmental Sounds", True)
+
+    # UI Properties
+    ui_enabled = checkbox_with_default("Enable UI Configuration", True)
+
+    # Decals
+    decals_enabled = checkbox_with_default("Enable Decals", True)
+
+    # Particle Effects
+    particles_enabled = checkbox_with_default("Enable Particle Effects", True)
+
+    # Advanced Game Object Properties Inputs
+    if color_enabled:
+        color = text_input_with_default("Enter Color (RGB, e.g., 255,0,0)", default_params["color"])
+    else:
+        color = None
+
+    if size_enabled:
+        size = text_input_with_default("Enter Size (X,Y,Z, e.g., 4,1,2)", default_params["size"])
+    else:
+        size = None
+
+    if position_enabled:
+        position = text_input_with_default("Enter Position (X,Y,Z, e.g., 0,10,0)", default_params["position"])
+    else:
+        position = None
+
+    if shape_enabled:
+        shape = selectbox_with_default("Select Shape", config["shapes"], default_params["shape"])
+    else:
+        shape = None
+
+    if material_enabled:
+        material = selectbox_with_default("Select Material", config["materials"], default_params["material"])
+    else:
+        material = None
+
+    if transparency_enabled:
+        transparency = text_input_with_default("Enter Transparency (0-1, e.g., 0.5)", default_params["transparency"])
+    else:
+        transparency = None
+
+    if reflectance_enabled:
+        reflectance = text_input_with_default("Enter Reflectance (0-1, e.g., 0.5)", default_params["reflectance"])
+    else:
+        reflectance = None
+
+    if rotation_speed_enabled:
+        rotation_speed = text_input_with_default("Enter Rotation Speed (e.g., 10)", default_params["rotation_speed"])
+    else:
+        rotation_speed = None
+
+    if bounce_height_enabled:
+        bounce_height = text_input_with_default("Enter Bounce Height (e.g., 10)", default_params["bounce_height"])
+    else:
+        bounce_height = None
+
+    if collision_enabled:
+        collision = selectbox_with_default("Enable Collision", config["collisions"], default_params["collision"])
+    else:
+        collision = None
+
+    if surface_enabled:
+        surface = selectbox_with_default("Select Surface Type", config["surfaces"], default_params["surface"])
+    else:
+        surface = None
+
+    if anchor_enabled:
+        anchor = selectbox_with_default("Enable Anchor", ["true", "false"], default_params["anchor"])
+    else:
+        anchor = None
+
+    if collision_group_enabled:
+        collision_group = selectbox_with_default("Select Collision Group", config["collision_groups"], default_params["collision_group"])
+    else:
+        collision_group = None
+
+    if custom_physics_enabled:
+        angular_velocity = text_input_with_default("Enter Angular Velocity (X,Y,Z, e.g., 0,0,0)", default_params["angular_velocity"])
+        linear_velocity = text_input_with_default("Enter Linear Velocity (X,Y,Z, e.g., 0,0,0)", default_params["linear_velocity"])
+        density = text_input_with_default("Enter Density", default_params["density"])
+    else:
+        angular_velocity = None
+        linear_velocity = None
+        density = None
+
+    if part_cframe_enabled:
+        part_cframe = text_input_with_default("Enter Part CFrame (e.g., CFrame.new(0, 50, -100))", default_params["part_cframe"])
+    else:
+        part_cframe = None
+
+    # Camera Properties Inputs
+    if camera_enabled:
+        camera_type = selectbox_with_default("Select Camera Type", config["camera_types"], default_params["camera_type"])
+        camera_position = text_input_with_default("Enter Camera Position (X,Y,Z, e.g., 0,50,-100)", default_params["camera_position"])
+        field_of_view = text_input_with_default("Enter Field of View", default_params["field_of_view"])
+    else:
+        camera_type = None
+        camera_position = None
+        field_of_view = None
+
+    if camera_interpolation_enabled:
+        camera_interpolation = checkbox_with_default("Enable Camera Interpolation", default_params["camera_interpolation"])
+    else:
+        camera_interpolation = None
+
+    # Lighting Properties Inputs
+    if lighting_enabled:
+        ambient_light = text_input_with_default("Enter Ambient Light (RGB, e.g., 128,128,128)", default_params["ambient_light"])
+        brightness = text_input_with_default("Enter Brightness", default_params["brightness"])
+        time_of_day = selectbox_with_default("Select Time of Day", config["times_of_day"], default_params["time_of_day"])
+        shadow_softness = text_input_with_default("Enter Shadow Softness (0-1, e.g., 0.5)", default_params["shadow_softness"])
+        shadow_color = text_input_with_default("Enter Shadow Color (RGB, e.g., 0,0,0)", default_params["shadow_color"])
+    else:
+        ambient_light = None
+        brightness = None
+        time_of_day = None
+        shadow_softness = None
+        shadow_color = None
+
+    if fog_enabled:
+        fog_start = text_input_with_default("Enter Fog Start Distance", default_params["fog_start"])
+        fog_end = text_input_with_default("Enter Fog End Distance", default_params["fog_end"])
+        fog_color = text_input_with_default("Enter Fog Color (RGB, e.g., 128,128,128)", default_params["fog_color"])
+    else:
+        fog_start = None
+        fog_end = None
+        fog_color = None
+
+    # Player Properties Inputs
+    if player_settings_enabled:
+        walk_speed = text_input_with_default("Enter Walk Speed", default_params["walk_speed"])
+        jump_power = text_input_with_default("Enter Jump Power", default_params["jump_power"])
+        health = text_input_with_default("Enter Health", default_params["health"])
+    else:
+        walk_speed = None
+        jump_power = None
+        health = None
+
+    if respawn_time_enabled:
+        respawn_time = text_input_with_default("Enter Respawn Time", default_params["respawn_time"])
+    else:
+        respawn_time = None
+
+    # Sound Properties Inputs
+    if sound_enabled:
+        background_music = text_input_with_default("Enter Background Music ID (e.g., rbxassetid://123456789)", default_params["background_music"])
+        sound_volume = text_input_with_default("Enter Sound Volume (e.g., 1)", default_params["sound_volume"])
+    else:
+        background_music = None
+        sound_volume = None
+
+    if sound_effects_enabled:
+        sound_effects = text_input_with_default("Enter Sound Effects ID (e.g., rbxassetid://987654321)", default_params["sound_effects"])
+    else:
+        sound_effects = None
+
+    if environmental_sounds_enabled:
+        environmental_sounds = text_input_with_default("Enter Environmental Sounds ID (e.g., rbxassetid://1122334455)", default_params["environmental_sounds"])
+    else:
+        environmental_sounds = None
+
+    # UI Properties Inputs
+    if ui_enabled:
+        ui_text = text_input_with_default("Enter UI Text", default_params["ui_text"])
+        ui_position = text_input_with_default("Enter UI Position (X,Y,Z, e.g., 0.5,-100,0.5,-25)", default_params["ui_position"])
+        ui_size = text_input_with_default("Enter UI Size (X,Y,Z, e.g., 0,200,0,50)", default_params["ui_size"])
+        ui_color = text_input_with_default("Enter UI Color (RGB, e.g., 255,255,255)", default_params["ui_color"])
+        ui_font = selectbox_with_default("Select UI Font", config["fonts"], default_params["ui_font"])
+        ui_text_size = text_input_with_default("Enter UI Text Size", default_params["ui_text_size"])
+    else:
+        ui_text = None
+        ui_position = None
+        ui_size = None
+        ui_color = None
+        ui_font = None
+        ui_text_size = None
+
+    # Decals Inputs
+    if decals_enabled:
+        texture_id = text_input_with_default("Enter Texture ID (e.g., rbxassetid://123456789)", default_params["texture_id"])
+        decal_position = selectbox_with_default("Select Decal Position", config["decal_positions"], default_params["decal_position"])
+    else:
+        texture_id = None
+        decal_position = None
+
+    # Particle Effects Inputs
+    if particles_enabled:
+        particle_texture = text_input_with_default("Enter Particle Texture ID (e.g., rbxassetid://123456789)", default_params["particle_texture"])
+        particle_lifetime = text_input_with_default("Enter Particle Lifetime (e.g., 1,2)", default_params["particle_lifetime"])
+        particle_rate = text_input_with_default("Enter Particle Rate (e.g., 100)", default_params["particle_rate"])
+        particle_speed = text_input_with_default("Enter Particle Speed (e.g., 10,20)", default_params["particle_speed"])
+        particle_size = text_input_with_default("Enter Particle Size (e.g., 1,2)", default_params["particle_size"])
+        particle_acceleration = text_input_with_default("Enter Particle Acceleration (X,Y,Z, e.g., 0,-10,0)", default_params["particle_acceleration"])
+        particle_rot_speed = text_input_with_default("Enter Particle Rotation Speed (e.g., 100,200)", default_params["particle_rot_speed"])
+        particle_color = text_input_with_default("Enter Particle Color (RGB, e.g., 255,255,255)", default_params["particle_color"])
+    else:
+        particle_texture = None
+        particle_lifetime = None
+        particle_rate = None
+        particle_speed = None
+        particle_size = None
+        particle_acceleration = None
+        particle_rot_speed = None
+        particle_color = None
+
+    custom_prompt_enabled = checkbox_with_default("Enable Custom Prompt", False)
+
+    if custom_prompt_enabled:
+        custom_prompt = st.text_area("Enter Custom Prompt")
+    else:
+        custom_prompt = None
+
+    return {
+        "color": color, "size": size, "position": position, "shape": shape, "material": material,
+        "transparency": transparency, "reflectance": reflectance, "rotation_speed": rotation_speed,
+        "bounce_height": bounce_height, "collision": collision, "surface": surface, "anchor": anchor,
+        "collision_group": collision_group, "angular_velocity": angular_velocity, "linear_velocity": linear_velocity,
+        "density": density, "part_cframe": part_cframe, "camera_type": camera_type, "camera_position": camera_position,
+        "field_of_view": field_of_view, "camera_interpolation": camera_interpolation, "ambient_light": ambient_light,
+        "brightness": brightness, "time_of_day": time_of_day, "shadow_softness": shadow_softness, "shadow_color": shadow_color,
+        "fog_start": fog_start, "fog_end": fog_end, "fog_color": fog_color, "walk_speed": walk_speed, "jump_power": jump_power,
+        "health": health, "respawn_time": respawn_time, "background_music": background_music, "sound_volume": sound_volume,
+        "sound_effects": sound_effects, "environmental_sounds": environmental_sounds, "ui_text": ui_text, "ui_position": ui_position,
+        "ui_size": ui_size, "ui_color": ui_color, "ui_font": ui_font, "ui_text_size": ui_text_size, "texture_id": texture_id,
+        "decal_position": decal_position, "particle_texture": particle_texture, "particle_lifetime": particle_lifetime,
+        "particle_rate": particle_rate, "particle_speed": particle_speed, "particle_size": particle_size, 
+        "particle_acceleration": particle_acceleration, "particle_rot_speed": particle_rot_speed, "particle_color": particle_color,
+        "custom_prompt": custom_prompt
+    }
+
+def generate_lua_script(params, api_key):
+    client = Groq(api_key=api_key)
+    
+    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"
+
+    # Request the script generation from Groq
+    response = client.create_lua_script(prompt=prompt)
+    return response['script']
+
+def main():
+    config = load_config()
+    user_input = get_user_input(config)
+
+    if st.button("Generate Script"):
+        api_key = config["api_key"]
+        lua_script = generate_lua_script(user_input, api_key)
+        st.code(lua_script, language='lua')
+
+if __name__ == "__main__":
+    main()