Spaces:
Sleeping
Sleeping
File size: 2,444 Bytes
6255f8a b8270b1 6255f8a be39a04 6255f8a be39a04 6255f8a be39a04 6255f8a be39a04 b8270b1 6255f8a b8270b1 6255f8a b8270b1 6255f8a b8270b1 6255f8a 3b45de9 6255f8a be39a04 6255f8a be39a04 6255f8a |
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 |
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):
# Generate the plot
print(history)
history_str = format_history(history)
response = plot_creator.create_plot(user_input, history_str)
print(response)
history[-1][1] = response[0]
history = history + [(None, f"Here is the code used to generate the plot:")]
history = history + [(None, f"{response[1]}")]
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] = "I will be using the file you uploaded to generate the plot. Please describe the plot you would like to generate."
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=["image", "video", "audio"])
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.launch()
|