Upload 2 files

app.py

@@ -1,18 +1,38 @@
 import gradio as gr
 from revChatGPT.V1 import Chatbot
 
+import argparse
+
 #You can setup login information here, or login in from UI
 
 # If you want to use Email/Password to login, put your account information here
-email = None
-password = None
+email = ""
+password = ""
 
-# If you have an access token, replace the None with your token 
-access_token = None
+# If you have an access token, put your access token here
+access_token = ""
 
 # If you have a session token, put your session token here
-session_token = None
+session_token = ""
+
 
+def get_args():
+    parser = argparse.ArgumentParser(description='Command line args.')
+    parser.add_argument(
+        '--no_markdown',
+        action='store_true',
+        help='Disable the markdown of the web UI.',)
+    return parser.parse_args()
+
+def is_google_colab():
+    try:
+        import google.colab
+        return True
+    except:
+        return False
+    
+chatbot = None
+    
 def configure_chatbot(method, info):
     
     if method=="Email/Password":
@@ -30,9 +50,12 @@ def configure_chatbot(method, info):
         config.update({"access_token": access_token})
     elif session_token:
         config.update({"session_token": session_token})
-
+    
     global chatbot
-    chatbot = Chatbot(config=config)
+    try:
+        chatbot = Chatbot(config=config)
+    except:
+        chatbot = None
 
 login_method = ['Email/Password',
                 'Access token',
@@ -41,8 +64,11 @@ login_method = ['Email/Password',
 
 def ask_bot(prompt):
     message = ""
-    for data in chatbot.ask(prompt):
-        message = data["message"]
+    if chatbot:
+        for data in chatbot.ask(prompt):
+            message = data["message"]
+    else:
+        message = "The chatbot is not set up properly! Try to login again."
     return parse_text(message)
 
 def parse_text(text):
@@ -56,31 +82,36 @@ def parse_text(text):
                 lines[i] = f'</code></pre>'
         else:
             if i>0:
+                line = line.replace("<", "&lt;")
+                line = line.replace(">", "&gt;")
                 lines[i] = '<br/>'+line.replace(" ", "&nbsp;")
     return "".join(lines)
 
-def chatgpt_clone(inputs, history):
+def chat_clone(inputs, history):
     history = history or []
     output = ask_bot(inputs)
     history.append((inputs, output))
     return history, history
 
-with gr.Blocks() as demo:
-    gr.Markdown("""<h1><center>ChatGPT BOT build by revChatGPT & Gradio</center></h1>
-    """)
-    gr.Markdown(
-        "#### Author: [dotmet](https://github.com/dotmet)  Github link:[ChatGPTWEB](https://github.com/dotmet/chatgpt_webui)")
-    gr.Markdown(
-        "I have used my own OpenAI account for this demo，you can skip Login and try chat.")
-    gr.Markdown(
-        "Duplicate this space and run for your own account: [demo](https://huggingface.co/spaces/dotmet/chatgpt_webui?duplicate=true).")
+if ((email and password) or access_token or session_token):
+    css = "style.css"
+else:
+    css = None
+
+with gr.Blocks(css=css) as demo:
+    
+    args = get_args()
+    
+    if not args.no_markdown:
+        gr.Markdown("""<h1><center>ChatGPT BOT build by revChatGPT & Gradio</center></h1>""")
 
     if not ((email and password) or access_token or session_token):
-        gr.Markdown("""<h2>Login to OpenAI</h2>""")
+        if not args.no_markdown:
+            gr.Markdown("""<h2>Login to OpenAI</h2>""")
         with gr.Row():
             with gr.Group():
                 method = gr.Dropdown(label="Login Method", choices=login_method)
-                info = gr.Textbox(placeholder="email password/access_token/session_token", label="Login Information (select login method first)")
+                info = gr.Textbox(placeholder="email password/access_token/session_token", label="Login Information (choose login method first)")
                 with gr.Row():
                     login = gr.Button("Login")
                     login.click(configure_chatbot, inputs=[method, info])
@@ -95,12 +126,18 @@ with gr.Blocks() as demo:
             method = "Session token"
             info = session_token
         configure_chatbot(method, info)
-
-    gr.Markdown("""<h2>Start Chatting ...</h2>""")
-    chatbot1 = gr.Chatbot()
-    message = gr.Textbox(placeholder="Chat here")
-    state = gr.State()
+    
+    if not args.no_markdown:
+        gr.Markdown("""<h2>Start Chatting ...</h2>""")
+        
+    chatbot1 = gr.Chatbot(elem_id="chatbot", show_label=False)
+    state = gr.State([])
+    message = gr.Textbox(placeholder="Chat here", label="Human: ")
+    message.submit(chat_clone, inputs=[message, state], outputs=[chatbot1, state])
+    message.submit(lambda :"", None, message)
+    
     submit = gr.Button("SEND")
-    submit.click(chatgpt_clone, inputs=[message, state], outputs=[chatbot1, state])
+    submit.click(chat_clone, inputs=[message, state], outputs=[chatbot1, state])
+    submit.click(lambda :"", None, message)
 
-    demo.launch(debug = True, share=False)
+    demo.launch(debug = True, share=is_google_colab())

style.css

@@ -0,0 +1,10 @@
+.gradio-container {
+    max-width: 100%;
+    max-height: 100%;
+}
+
+[id$=chatbot] > div{
+    border: 0;
+    height: 70vh;
+    overflow-y: auto;
+}