Update app.py

app.py

@@ -2,25 +2,31 @@
 import os
 os.system(f"pip install open-interpreter")
 os.system(f"pip install matplotlib")
-
 import json
 import gradio as gr
 from interpreter import interpreter
-
 import time
 import matplotlib
 matplotlib.use('Agg') 
 
+
+#get openaiapikey from environment variable
+openaiapikey = os.environ.get('OPENAI_API_KEY')
+
+#install open-interpreter package
+os.system(f"pip install open-interpreter")
+
 interpreter.auto_run = True
 interpreter.llm.model = "gpt-4-turbo"
 interpreter.custom_instructions = "First ask the user what they want to do. Based on the input, describe the next steps. If user agrees, proceed; if not, ask what they want next.If it is anything to display , always at the end open up the file."
 
 def update_bot(text, chatbot):
-    response = interpreter.chat(text,stream=True, display=False)
-    response = json_to_markdown(response)
-    chatbot.append((text, response))
+    response_json = interpreter.chat(text, stream=True, display=False)
+    formatted_response = json_to_markdown(response_json)
+    chatbot.append((text, formatted_response))
     return chatbot, ""
 
+
 def new_chat():
     interpreter.messages = []
     return [], ""
@@ -52,21 +58,13 @@ def create_chat_widget():
         return chatblock
 
 def json_to_markdown(json_data):
-    markdown_text = []
+    full_message = []
     for item in json_data:
-        if item['role'] != 'user':  # Skip user entries
-            if item['type'] == 'message':
-                # Check if 'content' key exists before accessing it
-                content = item.get('content', "No content available")
-                markdown_text.append(f"**{item['role'].capitalize()}:** \n{content}")
-            elif item['type'] == 'code':
-                # Assume 'content' key exists, consider adding a similar check if necessary
-                markdown_text.append(f"```{item['format']}\n{item['content']}\n```")
-            elif item['type'] == 'console':
-                # Check for 'content' in console type
-                content = item.get('content', "No content available")
-                markdown_text.append(f"```\n{content}\n```")
-    return "\n\n".join(markdown_text)
+        if item['role'] == 'assistant' and item['type'] == 'message':
+            content = item.get('content', " ")
+            full_message.append(content)  # Append the content to the list
+    return "**Assistant:**\n" + " ".join(full_message)  # Join all parts into one string and format as a single message
+
 
 with gr.Blocks() as demo:
     with gr.Tab("HEXON Chatbot Assignment"):