Spaces:
Sleeping
Sleeping
File size: 2,929 Bytes
02cd084 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
import gradio as gr
import os
from llm.gptPlotCreator import PlotCreator
plot_creator = PlotCreator()
def add_text(history, text):
history = history + [(text, None)]
return history, ""
def add_file(history, file):
history = history + [((file.name,), None)]
return history
def format_history(history):
return "\n".join([f"Human: {entry[0]}\nAI: {entry[1]}" for entry in history ])
def bot(history):
# Get the last input from the user
user_input = history[-1][0] if history and history[-1][0] else None
print(user_input)
# Check if it is a string
if isinstance(user_input, str):
history[-1][1] = "I am figuring out what data types are relevant for the plot...\n"
yield history
data_types_str = plot_creator.find_relevant_data_types(user_input)
history[-1][1] += "I am now generating a script to plot the data...\n"
yield history
plot_creator.create_plot(user_input, data_types_str)
history[-1][1] += "I am now running the script I just Generated...\n"
yield history
response = plot_creator.run_script()
history = history + [(None, f"Here is the code used to generate the plot:")]
history = history + [(None, f"{response[1]}")]
history = history + response[0]
yield history
else:
file_path = user_input[0]
plot_creator.set_logfile_name(file_path)
# get only base name
filename, extension = os.path.splitext(os.path.basename(file_path))
history[-1][0] = f"user uploaded file: {filename}{extension}"
history[-1][1] = "processing file..."
yield history
data_types = plot_creator.parse_mavlink_log()
history = history + [(None, f"I am done processing the file. Now you can ask me to generate a plot.")]
yield history
return history
with gr.Blocks() as demo:
gr.Markdown("# GPT MAVPlot\n\nThis web-based tool allows users to upload mavlink tlogs in which the chat bot will use to generate plots from. It does this by creating a python script using pymavlink and matplotlib. The output includes the plot and the code used to generate it. ")
chatbot = gr.Chatbot([], elem_id="chatbot").style(height=750)
with gr.Row():
with gr.Column(scale=0.85):
txt = gr.Textbox(
show_label=False,
placeholder="Enter text and press enter, or upload an image",
).style(container=False)
with gr.Column(scale=0.15, min_width=0):
btn = gr.UploadButton("📁", file_types=["file"])
txt.submit(add_text, [chatbot, txt], [chatbot, txt]).then(
bot, chatbot, chatbot
)
btn.upload(add_file, [chatbot, btn], [chatbot]).then(
bot, chatbot, chatbot
)
if __name__ == "__main__":
demo.queue()
demo.launch()
|