Update app.py

app.py

@@ -1,9 +1,5 @@
 import gradio as gr
 import openai
-import os
-
-current_dir = os.path.dirname(os.path.abspath(__file__))
-css_file = os.path.join(current_dir, "style.css")
 
 initial_prompt = "You are a helpful assistant."
 
@@ -25,30 +21,34 @@ def parse_text(text):
 
 def get_response(system, context, raw = False):
     openai.api_key = "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"
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=[system, *context],
+    response = openai.Completion.create(
+        engine="text-davinci-002",
+        prompt=f"{system}\n\n{context}",
+        temperature=0.7,
+        max_tokens=1024,
+        top_p=1,
+        frequency_penalty=0,
+        presence_penalty=0,
+        stop=None,
     )
     if raw:
         return response
     else:
-        statistics = f'This conversation Tokens usage【{response["usage"]["total_tokens"]} / 4096】 （ Question + above {response["usage"]["prompt_tokens"]}，Answer {response["usage"]["completion_tokens"]} ）'
-        message = response["choices"][0]["message"]["content"]
+        statistics = f'This conversation Tokens usage【{response["total_characters"]} / 2048】 （ Question + above {response["prompt"]["length"]}，Answer {response["choices"][0]["length"]} ）'
+        message = response.choices[0].text
 
-        message_with_stats = f'{message}'
-#         message_with_stats = markdown.markdown(message_with_stats)
+        message_with_stats = f'{message}\n\n{statistics}'
 
         return message, parse_text(message_with_stats)
-        #return message
 
 def predict(chatbot, input_sentence, system, context):
     if len(input_sentence) == 0:
         return []
-    context.append({"role": "user", "content": f"{input_sentence}"})
+    context.append(input_sentence)
 
     message, message_with_stats = get_response(system, context)
 
-    context.append({"role": "assistant", "content": message})
+    context.append(message)
 
     chatbot.append((input_sentence, message_with_stats))
 
@@ -58,9 +58,9 @@ def retry(chatbot, system, context):
     if len(context) == 0:
         return [], []
     message, message_with_stats = get_response(system, context[:-1])
-    context[-1] = {"role": "assistant", "content": message}
+    context[-1] = message
 
-    chatbot[-1] = (context[-2]["content"], message_with_stats)
+    chatbot[-1] = (context[-2], message_with_stats)
     return chatbot, context
 
 def delete_last_conversation(chatbot, context):
@@ -70,26 +70,11 @@ def delete_last_conversation(chatbot, context):
     context = context[:-2]
     return chatbot, context
 
-def reduce_token(chatbot, system, context):
-    context.append({"role": "user", "content": "请帮我总结一下上述对话的内容，实现减少tokens的同时，保证对话的质量。在总结中不要加入这一句话。"})
-
-    response = get_response(system, context, raw=True)
-
-    statistics = f'本次对话Tokens用量【{response["usage"]["completion_tokens"]+12+12+8} / 4096】'
-    optmz_str = markdown.markdown( f'好的，我们之前聊了:{response["choices"][0]["message"]["content"]}\n\n================\n\n{statistics}' )
-    chatbot.append(("请帮我总结一下上述对话的内容，实现减少tokens的同时，保证对话的质量。", optmz_str))
-    
-    context = []
-    context.append({"role": "user", "content": "我们之前聊了什么?"})
-    context.append({"role": "assistant", "content": f'我们之前聊了：{response["choices"][0]["message"]["content"]}'})
-    return chatbot, context
-
-
 def reset_state():
     return [], []
 
 def update_system(new_system_prompt):
-    return {"role": "system", "content": new_system_prompt}
+    return new_system_prompt
 
 title = """<h1 align="center">Tu întrebi și eu răspund.</h1>"""
 description = """<div align=center>
@@ -99,40 +84,15 @@ Will not describe your needs to ChatGPT？You Use [ChatGPT Shortcut](https://new
 </div>
 """
 
-with gr.Blocks() as demo:
-    gr.HTML(title)
-    chatbot = gr.Chatbot().style(color_map=("#A238FF", "#A238FF"))
-    context = gr.State([])
-    systemPrompt = gr.State(update_system(initial_prompt))
-
-    with gr.Row():
-        with gr.Column(scale=12):
-            txt = gr.Textbox(show_label=False, placeholder="Please enter any of your needs here").style(container=False)
-        with gr.Column(min_width=50, scale=1):
-            #submitBtn = gr.Button("🚀 Submit", variant="Primary")
-            submitBtn = gr.Button("🚀 Submit").style(css={"background-color": "#A238FF"})
-    with gr.Row():
-        emptyBtn = gr.Button("🧹 New conversation")
-        retryBtn = gr.Button("🔄 Resubmit")
-        delLastBtn = gr.Button("🗑️ Delete conversation")
-        #reduceTokenBtn = gr.Button("♻️ Optimize Tokens")
-
-    #newSystemPrompt = gr.Textbox(show_label=True, placeholder=f"Setting System Prompt...", label="Change System prompt").style(container=True)
-    #systemPromptDisplay = gr.Textbox(show_label=True, value=initial_prompt, interactive=False, label="Current System prompt").style(container=True)
-    
-    #gr.Markdown(description)
-    
-    txt.submit(predict, [chatbot, txt, systemPrompt, context], [chatbot, context], show_progress=True)
-    txt.submit(lambda :"", None, txt)
-    submitBtn.click(predict, [chatbot, txt, systemPrompt, context], [chatbot, context], show_progress=True)
-    submitBtn.click(lambda :"", None, txt)
-    emptyBtn.click(reset_state, outputs=[chatbot, context])
-    #newSystemPrompt.submit(update_system, newSystemPrompt, systemPrompt)
-    #newSystemPrompt.submit(lambda x: x, newSystemPrompt, systemPromptDisplay)
-    #newSystemPrompt.submit(lambda :"", None, newSystemPrompt)
-    retryBtn.click(retry, [chatbot, systemPrompt, context], [chatbot, context], show_progress=True)
-    delLastBtn.click(delete_last_conversation, [chatbot, context], [chatbot, context], show_progress=True)
-    #reduceTokenBtn.click(reduce_token, [chatbot, systemPrompt, context], [chatbot, context], show_progress=True)
-
-
-demo.launch()
+gr.Interface(
+    predict, 
+    inputs=["textbox", "text", "text", "text"],
+    outputs="text",
+    title=title,
+    description=description,
+    allow_flagging=False,
+    theme="huggingface",
+    layout="vertical",
+    examples=[["What's your name?", "I don't have a name. How can I assist you today?", "", ""]],
+    interpretation="default"
+).launch()