Update app.py

app.py

@@ -2,25 +2,25 @@ import gradio as gr
 import json
 import asyncio
 import os
-from EdgeGPT import Chatbot, ConversationStyle
 import re
+from EdgeGPT import Chatbot, ConversationStyle
 
 # Read cookie from local file
 with open('./cookies.json', 'r') as f:
     cookies = json.load(f)
-# If you choose to read the cookie.json file in the repository, you don't need to pass the cookies parameter to the function again
-# You can also delete the code corresponding to the gr.Tab("Cookies"): interface
+
+# Async function to get model reply
 async def get_model_reply(prompt, style, context=[]):
     # Combines the new question with a previous context
     context += [prompt]
-    # Cookies = json.loads(cookies)
-    # Given the most recent context (4096 characters)
-    # Continue the text up to 2048 tokens ~ 8192 characters
+    # Initialize chatbot with cookies
     bot = Chatbot(cookies=cookies)
+    # Limit the context to 4096 characters
     prompt2 = '\n\n'.join(context)[:4096]
     raw_data = await bot.ask(prompt2, conversation_style=style)
     await bot.close()
-    # Print(raw_data)
+
+    # Try to get response from raw data
     try:
         try:
             response = raw_data["item"]["messages"][1]["text"]
@@ -29,57 +29,53 @@ async def get_model_reply(prompt, style, context=[]):
         response = re.sub(r'\^', '', response)
         response = response.rstrip()
         context += [response]
-
-        # List of (user, bot) responses. We will use this format later
+        # List of (user, bot) responses
         responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
         return responses, context
     except:
         try:
+            # Check if conversation limit is reached
             if raw_data["item"]["throttling"]["numUserMessagesInConversation"] > raw_data["item"]["throttling"]["maxNumUserMessagesInConversation"]:
                 response = "> **Oops, I think we've reached the end of this conversation. Please reset the bot!**"
                 context += [response]
-
-                # List of (user, bot) responses. We will use this format later
                 responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
                 return responses, context
-
         except:
+            # Check if user is throttled
             if raw_data["item"]["result"]["value"] == "Throttled":
                 response = "> **Error: We're sorry, but you've reached the maximum number of messages you can send to Bing in a 24-hour period. Check back later!**"
                 context += [response]
-
-                # List of (user, bot) responses. We will use this format later
                 responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
                 return responses, context
 
-# Query = 'Which is the largest country by area in the world?'
-# Style="precise"
-# Responses, context =asyncio.run(get_model_reply(query,style,context=[]))
-#
-# Print(' ' + responses[-1][0])
-# Print(' ' + responses[-1][1])
+# Function to send user input to model and get reply
+def send(inputs, style, state):
+    # Run async function and get responses
+    responses, context = asyncio.run(get_model_reply(inputs, style, state))
+    # Append user input and bot response to chat interface
+    chatbot.append_message("user", inputs)
+    chatbot.append_message("bot", responses[-1][1])
+    # Update state
+    state.append(responses)
+
+# Define Gradio interface
 with gr.Blocks() as dialog_app:
-    # With gr.Tab("Cookies"):
-        # Cookies = gr.Textbox(lines=2, label="Enter cookies from bing.com")
-    # With gr.Tab("New Bing Chat"):
-        gr.Markdown("# A Simple Web to use New Bing Without Magic")
-        chatbot = gr.Chatbot()
-        state = gr.State([])
-        markdown = gr.Markdown(label="Output")
+    gr.Markdown("# A Simple Web to use New Bing Without Magic")
+    chatbot = gr.Chatbot()
+    state = gr.State([])
+    markdown = gr.Markdown(label="Output")
 
-        with gr.Row():
-            inputs = gr.Textbox(
-                label="Enter question",
-                placeholder="Enter text and press enter"
-            )
-            style = gr.Dropdown(label="Answer tendency", choices=["creative", "balanced", "precise"], multiselect=False,
-                                value="balanced", type="value")
+    # Input fields
+    with gr.Row():
+        inputs = gr.Textbox(label="Enter question", placeholder="Enter text and press enter")
+        style = gr.Dropdown(label="Answer tendency", choices=["creative", "balanced", "precise"], multiselect=False, value="balanced", type="value")
 
-        inputs.submit(get_model_reply, [inputs, style, state], [chatbot, state])
-        send = gr.Button("Send")
-        send.click(get_model_reply, [inputs, style, state], [chatbot, state])
+    # Set 'send' as submit function for input fields
+    inputs.submit(send, [inputs, style, state])
+    send = gr.Button("Send")
+    # Set 'send' as click function for 'Send' button
+    send.click(send, [inputs, style, state])
 
-# Launches the app in a new local port
+# Launch Gradio app
 dialog_app.launch()
-# Set a password for the website to prevent abuse
-# Dialog_app.launch(auth=("admin", "pass1234"))
+