more dev

.gitignore

@@ -1,2 +1,4 @@
 .env
-*.pyc
+*.pyc
+plot.py
+plot.png

README.md

@@ -0,0 +1,37 @@
+# GPT_MAVPlot
+
+MAVPlot is a Python-based project which uses Gradio as an interface and GPT-X powered by OpenAI as a chatbot to generate and plot MAVLink data. It provides an easy-to-use, chatbot-like interface for users to describe the plot they would like to generate.
+
+## Installation
+
+1. Clone the repository:
+
+```shell
+git clone https://github.com/yourusername/mavplot.git
+```
+
+2. Install the requirements:
+
+```shell
+pip install -r requirements.txt
+```
+
+## Usage
+
+After installing all dependencies, run the main script using:
+
+```shell
+python llm_plot.py
+```
+
+A web-based Gradio interface will launch. You can then input the description of the plot you would like to generate in the textbox, or upload a file.
+
+The chatbot will process your request and generate the corresponding plot, which will be displayed in the chat interface.
+
+## Contributing
+
+Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change. Please make sure to update tests as appropriate.
+
+## License
+
+[MIT](https://choosealicense.com/licenses/mit/)

llm/gptPlotCreator.py

@@ -44,8 +44,8 @@ class PlotCreator:
 
     @staticmethod
     def extract_code_snippets(text):
-        pattern = r'```(.*?)```'
-        snippets = re.findall(pattern, text, re.DOTALL)
+        pattern = r'```.*?\n(.*?)```'
+        snippets = re.findall(pattern, text, re.DOTALL | re.MULTILINE)
         if len(snippets) == 0:
             snippets = [text]
         return snippets
@@ -76,6 +76,7 @@ class PlotCreator:
 
         # run the script 
         os.system("python plot.py")
+        return code
 
     def create_plot(self, human_input):
         file = "data/2023-01-04 20-51-25.tlog"
@@ -95,9 +96,9 @@ class PlotCreator:
             subprocess.check_output(["python", "plot.py"], stderr=subprocess.STDOUT)
         except subprocess.CalledProcessError as e:
             print(e.output.decode())
-            self.attempt_to_fix_sctript("plot.py", e.output.decode())
+            code = self.attempt_to_fix_sctript("plot.py", e.output.decode())
         except Exception as e:
             print(e)
-            self.attempt_to_fix_sctript("plot.py", str(e))
+            code = self.attempt_to_fix_sctript("plot.py", str(e))
 
-        return ("plot.png", None)
+        return [("plot.png", None), code[0]]

llm_plot.py

@@ -18,15 +18,16 @@ def bot(history):
     # Check if it is a string
     if isinstance(user_input, str):
         # Generate the plot
-        img = plot_creator.create_plot(user_input)
-        response = img
+        response = plot_creator.create_plot(user_input)
     else:
         response = "**That's cool!**"
 
-    history[-1][1] = ('plot.png', None)
+    history[-1][1] = response[0]
+    history = history + [(None, f"Here is the code used to generate the plot:\n```\n{response[1]}```")]
     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():