Update app.py

app.py

@@ -1,26 +1,15 @@
-import os
-os.system("python -m pip install --upgrade pip") 
 import openai
 import gradio as gr
 from gradio.components import Audio, Textbox
+import os
 import re
 import tiktoken
-
+from transformers import GPT2Tokenizer
 import whisper
+import pandas as pd
 from datetime import datetime, timezone, timedelta
 import notion_df
 import concurrent.futures
-from IPython.core.display import HTML
-os.system("pip install transformers>=4.0.0 pandas numpy spacy")
-from transformers import GPT2Tokenizer
-import pandas as pd
-import numpy as np
-import spacy
-
-
-from spacy import displacy
-
-nlp = spacy.load("en_core_web_sm")
 
 # Define the tokenizer and model
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2-medium')
@@ -137,32 +126,14 @@ def transcribe(audio, text):
     df = pd.DataFrame([chat_transcript])
     notion_df.upload(df, 'https://www.notion.so/YENA-be569d0a40c940e7b6e0679318215790?pvs=4', title=str(published_date), api_key=API_KEY)
 
-    # Colorize the system message text
-    colorized_system_message = colorize_text(system_message['content'])
-
-    # Return the colorized chat transcript
-    return colorized_system_message
-
+    # Return the chat transcript
+    return system_message['content']
     
-def colorize_text(text):
-    doc = nlp(text)
-    colorized_text = ""
-
-    for token in doc:
-        if token.ent_type_:
-            colorized_text += f'<span style="background-color: yellow;">{token.text_with_ws}</span>'
-        elif token.pos_ in {'NOUN'}:
-            colorized_text += f'<span style="color: blue;">{token.text_with_ws}</span>'
-        else:
-            colorized_text += token.text_with_ws
-
-    return colorized_text
-
-
 # Define the input and output components for Gradio
 audio_input = Audio(source="microphone", type="filepath", label="Record your message")
 text_input = Textbox(label="Type your message", max_length=4096)
-output_text = Markdown()
+output_text = gr.outputs.Textbox(label="Response")
+output_audio = Audio()
 
 # Define the Gradio interface
 iface = gr.Interface(
@@ -177,4 +148,4 @@ iface = gr.Interface(
     )
 
 # Run the Gradio interface
-iface.launch(enable_queue=True)
+iface.launch()