clenaup

app.py

@@ -99,10 +99,9 @@ def get_response(prompt: str, *kargs):
     elif prompt_lower.startswith('plot'):
         p = prompt_lower.split('plot ')[1]
         on_render = st.write
-        response = from_gpt(p, True, llm)
-    else:
+        response = from_gpt(p, plot=True)
         on_render = st.write
-        response = from_gpt(prompt, False, llm)
+        response = from_gpt(prompt, plot=False)
 
     return (response, on_render)