Upload folder using huggingface_hub

app.py

@@ -1,11 +1,14 @@
 import openai
+import re
 import gradio as gr
 from prompts import prompt_chat_response
 from fewshot import FewShot4UAVs
+from flask import requests
 
 fewshot = FewShot4UAVs()
 
-"""def send_command(command, location=None):
+
+def send_command(command, location=None):
     url = f"http://url/{command}"
     data = {}
     if location:
@@ -20,61 +23,40 @@ def parse_command(text):
     if len(tokens) >= 2:
         return tokens[1]
     else:
-        return 'none'"""
+        return 'none'
+
 
+def transcribe(audio):
+    messages = [
+        {"role": "system",
+         "content": prompt_chat_response}
+    ]
 
-def transcribe_audio(audio):
     audio_file = open(audio, "rb")
     transcript = openai.Audio.transcribe("whisper-1", audio_file)
-    return transcript["text"]
-
 
-def append_user_transcript(messages, user_transcript):
-    messages.append({"role": "user", "content": user_transcript, "name": "Operator"})
+    original_transcript = transcript["text"]
+    messages.append({"role": "user", "content": original_transcript, "name": "Operator"})
 
-
-def append_formatted_command(messages, user_transcript):
-    formatted_command_text = fewshot.get_command(user_transcript)
+    formatted_command_text = fewshot.get_command(original_transcript)
     messages.append({"role": "function", "content": formatted_command_text, "name": "UAV"})
-    return formatted_command_text
 
+    # turn on the below to invoke API
+    # command = parse_command(formatted_command_text)
+    # send_command(command)
 
-def append_uav_response(messages):
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         messages=messages
     )
+
     uav_response = response["choices"][0]["message"]["content"]
     messages.append({"role": "assistant", "content": uav_response, "name": "Assistant"})
-    return uav_response
-
 
-def build_chat_transcript(messages):
     chat_transcript = ""
     for message in messages:
         if message['role'] != 'system':
             chat_transcript += f"{message['name']}: {message['content']} \n\n"
-    return chat_transcript
-
-
-def transcribe(audio):
-    messages = [
-        {"role": "system",
-         "content": prompt_chat_response}
-    ]
-
-    user_transcript = transcribe_audio(audio)
-    append_user_transcript(messages, user_transcript)
-
-    formatted_command_text = append_formatted_command(messages, user_transcript)
-
-    # turn on the below to invoke API
-    # command = parse_command(formatted_command_text)
-    # send_command(command)
-
-    uav_response = append_uav_response(messages, formatted_command_text)
-
-    chat_transcript = build_chat_transcript(messages)
 
     return chat_transcript
 
@@ -100,4 +82,4 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
                     "Land now.")
 
 demo.queue()
-demo.launch()
+demo.launch()