Update app.py

app.py

@@ -1,3 +1,5 @@
+import gradio as gr
+import os
 import numpy as np
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 import requests
@@ -10,55 +12,50 @@ class SecondLifeNavigator:
         self.corrade_api_key = corrade_api_key
 
     def send_command_to_corrade(self, corrade_endpoint, command, parameters):
-        # Construct the command data
         command_data = {
             "command": command,
             "group": "e269893f-a570-0087-930e-6ba2a0b77f9c",
-            "password": os.environ.get('pass')
+            "password": self.corrade_api_key,
         }
         command_data.update(parameters)
 
-        # Send the command to Corrade
         try:
             response = requests.post(corrade_endpoint, json=command_data)
             if response.status_code == 200:
-                print(f"Command {command} executed successfully.")
-                return response.json()  # Assuming Corrade returns JSON
+                return f"Command {command} executed successfully.", response.json()
             else:
-                print(f"Failed to execute command {command}: HTTP {response.status_code}")
-                return None
+                return f"Failed to execute command {command}: HTTP {response.status_code}", None
         except Exception as e:
-            print(f"Error sending command to Corrade: {e}")
-            return None
+            return f"Error sending command to Corrade: {e}", None
 
     def generate_action_sequence(self, current_state):
-        print("Inside Action Sequence...")
         inputs = self.tokenizer.encode(current_state, return_tensors='pt')
         outputs = self.model.generate(inputs, max_length=40, num_return_sequences=1)
         action_sequence = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
         return action_sequence
 
-    def execute_actions(self, corrade_endpoint, action_sequence):
-        actions = action_sequence.split(';')
-        for action in actions:
-            if action.strip() == "move_forward":
-                self.send_command_to_corrade(corrade_endpoint, "move", {"direction": "forward"})
-            elif action.strip() == "interact_with_object":
-                self.send_command_to_corrade(corrade_endpoint, "interact", {"item": "9e7c8164-b633-c0f4-23f0-916a0144d836"})
-            # Implement other actions as needed
+    # Simplify run method for Gradio compatibility
+    def run(self, current_state, corrade_endpoint):
+        action_sequence = self.generate_action_sequence(current_state)
+        # Removed automatic command execution for safety and demonstration purposes
+        return action_sequence
 
-    def run(self, corrade_endpoint):
-        try:
-            current_states = ["I am at the starting point.", "I see an object nearby.", "I am near the DJ equipment."]
-            for state in current_states:
-                print(f"Current state: {state}")
-                action_sequence = self.generate_action_sequence(state)
-                print(f"Generated action sequence: {action_sequence}")
-                self.execute_actions(corrade_endpoint, action_sequence)
-        except Exception as e:
-            print(f"An error occurred: {e}")
+def create_interface():
+    def navigate(current_state, corrade_endpoint):
+        sl_navigator = SecondLifeNavigator()
+        action_sequence = sl_navigator.run(current_state, corrade_endpoint)
+        return action_sequence
+
+    interface = gr.Interface(
+        fn=navigate,
+        inputs=[gr.inputs.Textbox(lines=2, placeholder="Enter current state..."),
+                gr.inputs.Textbox(label="Corrade Endpoint URL")],
+        outputs=[gr.outputs.Textbox(label="Generated Action Sequence")],
+        title="Second Life Navigator",
+        description="Generates an action sequence based on the current state for navigating in Second Life.",
+    )
+
+    interface.launch()
 
 if __name__ == "__main__":
-    corrade_endpoint = os.environ.get('url')  # Define the Corrade endpoint here
-    sl_navigator = SecondLifeNavigator()
-    sl_navigator.run(corrade_endpoint)
+    create_interface()