Update app.py

app.py

@@ -1,16 +1,29 @@
+import re
 import openai
 import gradio as gr
 
+# Define a regular expression to match Python code blocks
+code_pattern = re.compile(r"```python\n(.*?)\n```", re.DOTALL)
+
 # Define the chat function
 def chat(api_key, model, message):
     # Check if an API key has been provided
     if api_key is None:
         return "Please enter your OpenAI API key and try again."
     
+    # Extract code blocks from the message
+    code_blocks = extract_code_blocks(message)
+    
+    # Replace code blocks with highlighted versions
+    highlighted_message = message
+    for code in code_blocks:
+        highlighted_code = f'<span style="background-color: #FFFF00;">{code}</span>'
+        highlighted_message = highlighted_message.replace(f'```python\n{code}\n```', highlighted_code)
+    
     # Set up the OpenAI API request
     response = openai.Completion.create(
         engine=model,
-        prompt=message,
+        prompt=highlighted_message,
         max_tokens=1024,
         n=1,
         stop=None,
@@ -21,7 +34,20 @@ def chat(api_key, model, message):
     # Extract the bot's response from the API request
     bot_response = response.choices[0].text.strip()
     
-    return bot_response
+    # Highlight code blocks in the bot response
+    highlighted_bot_response = bot_response
+    for code in code_blocks:
+        highlighted_code = f'<span style="background-color: #FFFF00;">{code}</span>'
+        highlighted_bot_response = highlighted_bot_response.replace(code, highlighted_code)
+    
+    return highlighted_bot_response
+
+# Define a function to extract code blocks from a string
+def extract_code_blocks(text):
+    code_blocks = []
+    for match in code_pattern.finditer(text):
+        code_blocks.append(match.group(1))
+    return code_blocks
 
 # Define the Gradio interface
 api_key_input = gr.inputs.Textbox(label="OpenAI API Key", default=None)
@@ -31,7 +57,7 @@ model_input = gr.inputs.Dropdown(
     default="text-davinci-003",
 )
 message_input = gr.inputs.Textbox(label="Enter your message here")
-output = gr.outputs.Textbox(label="Bot response")
+output = gr.outputs.HTML(label="Bot response")
 
 chat_button = gr.Interface(
     fn=chat,
@@ -39,7 +65,7 @@ chat_button = gr.Interface(
     outputs=output,
     title="OpenAI Chatbot",
     description="Enter your message below to chat with an AI",
-    theme="huggingface",
+    theme="compact",
     layout="vertical",
     allow_flagging=False,
     allow_screenshot=False,