improvements on brain service manage.sh

brain/brain_base.py

@@ -4,7 +4,7 @@ import os
 
 from loguru import logger
 
-from commander.commands import CommandHandler
+from brain.commands import CommandHandler
 
 
 class BaseBrain:

brain/service_fastapi.py

@@ -1,34 +0,0 @@
-import sys
-
-from fastapi import FastAPI
-import uvicorn
-
-from brain_openai import CloudChatBrain
-
-
-
-#def start_application():
-#    app = FastAPI()
-#    # app.include_router(api_router)
-#    # brain = CloudChatBrain()
-#    return app
-
-
-
-
-if __name__ == "__main__":
-    if len(sys.argv) == 1: 
-        port = 8890
-    else:
-        port = int(sys.argv[1])
-
-    app = FastAPI()
-
-    @app.post("/command/")
-    def post_command(payload: dict):
-        prompt = payload.get("prompt")
-        return {"response": prompt}
-
-    
-    #app = start_application()
-    uvicorn.run(app, host="0.0.0.0", port=port, log_level="debug")

brain/service_gradio_nicetry_1.py

@@ -1,37 +0,0 @@
-import gradio as gr
-import requests
-
-
-conversation = []
-
-def generate_response(prompt, params):
-    #response = requests.get(f"http://localhost:8000/command/{prompt}").json()
-    response = f"I got {prompt}"
-    conversation.append({"prompt": prompt, "response": response})
-    print(params)
-    #params["conversation_history"].description = f"Conversation History:\n\n{populate_chat_history(conversation)}"
-    return f"I got: {prompt}"
-
-def populate_chat_history(conversation):
-    history = ""
-    for i, chat in enumerate(conversation):
-        history += f"User: {chat['prompt']}\nBot: {chat['response']}\n\n"
-    return history
-
-iface = gr.Interface(
-    fn=generate_response,
-    inputs=["text"],
-    outputs=["text"],
-    parameters=[
-        {
-            "type": "textbox", 
-            "key": "conversation_history", 
-            "label": "Conversation:", 
-            "default": ""
-        }
-    ],
-    title="fly my drone by talking",
-    description="what should I do now?\n(enter q for emergency)"
-)
-
-iface.launch(server_name="0.0.0.0", server_port=8890, debug=True)

manage.sh

@@ -59,16 +59,18 @@ start_codeserver(){
 }
 
 start_commander_service() {
-	venv/bin/python commander/commander.py $1
+	venv/bin/python commander/service.py $1
 }
 
 start_brain_service() {
-    venv/bin/python -m gradio brain/service.py $1
+  commander_host=$1
+  port=$2
+  ENV_FOR_DYNACONF=$commander_host venv/bin/python -m gradio brain/service.py $port
 }
 
 talk_to_drone_via_cli() {
-    commander_host=$1
-    ENV_FOR_DYNACONF=$commander_host python brain/cli.py
+  commander_host=$1
+  ENV_FOR_DYNACONF=$commander_host python brain/cli.py
 }
 
 kill_everything() {
@@ -172,7 +174,7 @@ elif [ "$1" == "stop-commander" ]; then
 	fi
 
 elif [ "$1" == "start-brain" ]; then
-    start_brain_service $2 > logs/brain.log 2>&1 &
+    start_brain_service $2 $3 > logs/brain.log 2>&1 &
     brain_pid=$!
     echo "started brain service with PID brain_pid"
     echo brain_pid > $pids_dir/brain_pid.txt
@@ -211,8 +213,8 @@ else
   - get-/ kill-dhcp
   - start-/ stop-jupyter (8888)
   - start-/ stop-cs (8888)
-  - start-/ stop-commander [port (8889)]
-  - start-/ stop-brain [port (8890)]
+  - start-/ stop-commander port (8889)
+  - start-/ stop-brain [host (tuncel/commander)]  port (8890)
   - start-clitalk [host (tuncel/commander)]
   - turn-off
   - prepare-/ finish-flight"