Tidy

README.md

@@ -1,5 +1,5 @@
 ---
-title: Ethan Mollick - AI Demo
+title: Ethan Mollick
 emoji: 🏢
 colorFrom: purple
 colorTo: green

app.py

@@ -1,5 +1,4 @@
 import os
-import random
 import re
 import gradio as gr
 import openai
@@ -23,7 +22,6 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 def ai_complete(messages):
     response = openai.ChatCompletion.create(
-        #model = "gpt-3.5-turbo",
         model = "gpt-4",
         messages = messages,
         temperature=0.9,
@@ -35,14 +33,10 @@ def ai_complete(messages):
 
 def process_message(msg, history, role = "user"):
     history = history or [{"role": "system", "content": system_prompt}]
-    
     history.append({"role":role, "content":msg})
-    
     output = ai_complete(history)
     history.append({"role": output.role, "content": output.content})
-
     parsed_history, feedback, plan, grade = parse_history(history)
-
     return gr.update(value=""), parsed_history, feedback, plan, grade, history
 
 def parse_history(history):
@@ -78,10 +72,6 @@ def parse_history(history):
     return tuples, ai_feedback, ai_plan, ai_grade
 
 def find_tag_content(content, tag1, tag2):
-    tag1 = tag1.replace('[', '\[')
-    tag1 = tag1.replace(']', '\]')
-    tag2 = tag2.replace('[', '\[')
-    tag2 = tag2.replace(']', '\]')
     result = re.search(tag1 + '(.*)' + tag2, content, re.DOTALL)
     if result: return result.group(1).strip()
     return ""
@@ -90,21 +80,14 @@ def set_visible_if_interesting(component: gr.Textbox):
     return gr.update(visible=(component != ""))
 
 with gr.Blocks(theme=gr.themes.Soft(), css="main.css", title="Demo") as block:
-    
-    #gr.Markdown("""<h1><center>Demo</center></h1>""")
-
     state = gr.State()
     chatbot = gr.Chatbot(label="AI-VC", elem_id="chatbot")
-    message = gr.Textbox(label="User", elem_id="user_input")
-    
+    message = gr.Textbox(label="User", elem_id="user_input")    
     feedback = gr.Textbox(label="Feedback", visible=False, interactive=False, lines=4)
     plan = gr.Textbox(label="Plan", visible=False, interactive=False, lines=3)
     grade = gr.Textbox(label="Grade", visible=False, interactive=False, lines=1)
-    
     feedback.change(set_visible_if_interesting, feedback, feedback) 
     plan.change(set_visible_if_interesting, plan, plan) 
     grade.change(set_visible_if_interesting, grade, grade) 
-   
     message.submit(process_message, inputs=[message, state], outputs=[message, chatbot, feedback, plan, grade, state])
-
     block.launch(share=False)