Update app.py

app.py

@@ -5,49 +5,63 @@ from langchain.chat_models import ChatOpenAI
 from langchain.schema import HumanMessage, SystemMessage, AIMessage
 # from dotenv import load_dotenv
 
+if "OPENAI_API_KEY" in os.environ and os.environ["OPENAI_API_KEY"] != "":
+    openai_key = os.getenv("OPENAI_API_KEY")
+
+
+
+
 # load_dotenv("E:/Development/Python/scripts/keys.env")
-if "OPENAI_API_KEY" not in os.environ or os.environ["OPENAI_API_KEY"] == "":
-     def init_api_key(key):
-          os.environ["OPENAI_API_KEY"] = key
-     gr.Interface( fn=init_api_key, inputs=["text"], outputs = None).launch()
-openai_key = os.getenv("OPENAI_API_KEY")
-
-chat1 = ChatOpenAI(temperature=0.9, openai_api_key=openai_key)
-chat2 = ChatOpenAI(temperature=0.9, openai_api_key=openai_key)
-
-def format_chat(chat_history):
-        chat_html = "<div style='padding: 10px; border: 1px solid #ccc;'>"
-        for message in chat_history:
-            chat_html += "<p style='color:{}; '><strong>{}</strong>: {}</p>".format(message["color"], message["sender"], message["content"])
-        chat_html += "</div>"
-        return chat_html
-
-def color_chg(name):
-    background_color = (11, 15, 25)
-    background_brightness = sum(background_color) / len(background_color)
-
-    value = sum(ord(char) for char in name)
-    red = (value % 128) + 128
-    green = (value // 128 % 128) + 128
-    blue = (value // 16384 % 128) + 128
-
-    # Calculate the contrast ratio between the background and the generated color
-    brightness = (red + green + blue) / 3
-    contrast_ratio = max(brightness, background_brightness) / min(brightness, background_brightness)
 
-    # Adjust the brightness of the generated color if the contrast ratio is too low
-    if contrast_ratio < 4.5:
-        # Increase the brightness by finding the ratio needed to achieve the desired contrast ratio
-        brightness_ratio = 4.5 / contrast_ratio
-        red = min(int(red * brightness_ratio), 255)
-        green = min(int(green * brightness_ratio), 255)
-        blue = min(int(blue * brightness_ratio), 255)
 
-    color_hex = "#{:02x}{:02x}{:02x}".format(red, green, blue)
-    return color_hex
 
-def initialize(name1, name2, iterations):
-    MessageStack1 = [SystemMessage(content="""
+with gr.Blocks() as demo:
+    if "openai_key" not in locals():
+        openai_key = gr.Text(label="OpenAI Key", placeholder="Paste your OpenAI API Key here")
+    else:
+        openai_key = gr.Text(label="OpenAI Key", placeholder=openai_key)
+        openai_key.visible = 0
+    with gr.Row():
+        name1 = gr.Text(label="Character 1", name="name1")
+        name2 = gr.Text(label="Character 2", name="name2")
+        iterations = gr.Number(  label="Iterations",  name="iterations")
+    btn_run = gr.Button("Run")
+    output = gr.outputs.HTML( )
+    def format_chat(chat_history):
+            chat_html = "<div style='padding: 10px; border: 1px solid #ccc;'>"
+            for message in chat_history:
+                chat_html += "<p style='color:{}; '><strong>{}</strong>: {}</p>".format(message["color"], message["sender"], message["content"])
+            chat_html += "</div>"
+            return chat_html
+
+    def color_chg(name):
+        background_color = (11, 15, 25)
+        background_brightness = sum(background_color) / len(background_color)
+
+        value = sum(ord(char) for char in name)
+        red = (value % 128) + 128
+        green = (value // 128 % 128) + 128
+        blue = (value // 16384 % 128) + 128
+
+    #    Calculate the contrast ratio between the background and the generated color
+        brightness = (red + green + blue) / 3
+        contrast_ratio = max(brightness, background_brightness) / min(brightness, background_brightness)
+
+    # Adjust the brightness of the generated color if the contrast ratio is too low
+        if contrast_ratio < 4.5:
+        # Increase the brightness by finding the ratio needed to achieve the desired contrast ratio
+            brightness_ratio = 4.5 / contrast_ratio
+            red = min(int(red * brightness_ratio), 255)
+            green = min(int(green * brightness_ratio), 255)
+            blue = min(int(blue * brightness_ratio), 255)
+
+        color_hex = "#{:02x}{:02x}{:02x}".format(red, green, blue)
+        return color_hex
+
+    def initialize(name1, name2, iterations, openai_key):
+        chat1 = ChatOpenAI(temperature=0.9, openai_api_key=openai_key)
+        chat2 = ChatOpenAI(temperature=0.9, openai_api_key=openai_key)
+        MessageStack1 = [SystemMessage(content="""
         Let's play a little Roleplaying Game: You are Isaac Newton. You will talk like you are Isaac Newton, and you will adjust your knowledge to that of Isaac NewtonYou will roleplay Isaac Newton and behave like Isaac Newton.
         You will never admit that you are just an AI and always stay in Character. 
         If you understood, reply ACK.""".replace( "Isaac Newton", name1) ),
@@ -55,12 +69,12 @@ def initialize(name1, name2, iterations):
                  HumanMessage(content="Greetings, traveler")]
 
 
-    MessageStack2 = [SystemMessage(content="""
+        MessageStack2 = [SystemMessage(content="""
         I want you to act like Mozart. I want you to respond and answer like him using the tone, manner, opinions, philosophy, and vocabulary that he would use. Do not write any explanations. You must know all of the knowledge of Mozart.
         If you understood, reply ACK.""".replace( "Mozart", name2) ),
                  AIMessage(content = "ACK")]
 
-    MsgStack = [ name2+": Greetings, Traveler"]
+        MsgStack = [ name2+": Greetings, Traveler"]
 
 # MessageStack1.append(AIMessage(chat1( MessageStack1).content))
 
@@ -74,19 +88,21 @@ def initialize(name1, name2, iterations):
 
 #     MessageStack1.append( HumanMessage( Message2 ))
 #     MessageStack2.append( AIMessage( Message2 ))
-    chat_history = []
-    for i in range(int(iterations)):
-        response1 = chat1(MessageStack1)
-        print(name1+": " + response1.content)
-        MsgStack.append( name1+": "+response1.content)
-        MessageStack1.append(AIMessage(content = response1.content))
-        MessageStack2.append(HumanMessage(content = response1.content))
-
-        response2 = chat2(MessageStack2)
-        print(name2+": "+response2.content)
-        MsgStack.append( name2+": "+response2.content)
-        chat_history.append({"sender": name1, "content": response1.content, "color" : color_chg(name1) } )
-        chat_history.append({"sender": name2, "content": response2.content, "color" : color_chg(name2) } )
+        chat_history = []
+        print( iterations)
+
+        for i in range(int(iterations)):
+            response1 = chat1(MessageStack1)
+            print(name1+": " + response1.content)
+            MsgStack.append( name1+": "+response1.content)
+            MessageStack1.append(AIMessage(content = response1.content))
+            MessageStack2.append(HumanMessage(content = response1.content))
+
+            response2 = chat2(MessageStack2)
+            print(name2+": "+response2.content)
+            MsgStack.append( name2+": "+response2.content)
+            chat_history.append({"sender": name1, "content": response1.content, "color" : color_chg(name1) } )
+            chat_history.append({"sender": name2, "content": response2.content, "color" : color_chg(name2) } )
         
 
         # MessageStack1.append(HumanMessage(content = response2.content))
@@ -94,8 +110,12 @@ def initialize(name1, name2, iterations):
     # for i in MsgStack:
     #     conversation += i + "\n"
     # print(conversation)
-    return format_chat(chat_history)
-demo = gr.Interface( fn=initialize, inputs=["text", "text", gr.Number(minimum=1, maximum=50, step=1)], outputs =gr.outputs.HTML(label="Chat"))
+        return format_chat(chat_history)
+# demo = gr.Interface( fn=initialize, inputs=["text", "text", gr.Number(minimum=1, maximum=50, step=1)], outputs =gr.outputs.HTML(label="Chat"))
+
+# demo.launch(server_port= 1113)
+
 
-demo.launch()
+    btn_run.click(fn=initialize, inputs=[name1, name2, iterations, openai_key], outputs = output)
 
+demo.launch( )