Update app.py

app.py

@@ -2,54 +2,72 @@ import os
 import openai
 import gradio as gr
 
-#if you have OpenAI API key as an environment variable, enable the below
-#openai.api_key = os.getenv("OPENAI_API_KEY")
-
-#if you have OpenAI API key as a string, enable the below
-openai.api_key = "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"
+openai.api_key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxx"
 
 start_sequence = "\nAI:"
 restart_sequence = "\nHuman: "
 
-#prompt = "The following is a conversation with an AI assistant. The assistant is helpful, creative, clever, and very friendly.\n\nHuman: Hello, who are you?\nAI: I am an AI created by OpenAI. How can I help you today?\nHuman: "
-prompt = "You are TransGPT, a specialist in the field of transportation.The specialist supports the understanding and reply of Chinese.\n\nHuman: Hello, who are you?\nAI: I am an AI created by OpenAI. How can I help you today?\nHuman: "
-def openai_create(prompt):
-
-    response = openai.Completion.create(
-    model="text-davinci-003",
-    prompt=prompt,
-    temperature=0.9,
-    max_tokens=150,
-    top_p=1,
-    frequency_penalty=0,
-    presence_penalty=0.6,
-    stop=[" Human:", " AI:"]
-    )
+prompt_txt = "输入你的问题"
+prompt_img = "输入你的图像提示词"
+
+messages = [
+    {"role": "system", "content": "you are a very useful assistant"},
+]
 
-    return response.choices[0].text
+def openai_create(input):
+    user_message = {"role": "user", "content": input}
+    messages.append(user_message)
+    completion = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo-16k",
+        messages=messages,
+    )
+    completion = completion.choices[0].message.content
+    messages.append({"role": "assistant", "content": completion})
 
+    return completion
 
+def chatgpt_image(input, image_size, history,image_input):
+    completion = openai.Image.create(
+        prompt=input,
+        n=1,
+        size=image_size
+    )
+    return completion.data[0]['url'], history, ""
 
-def chatgpt_clone(input, history):
+def chatgpt_chat(input, history,input_text):
     history = history or []
-    s = list(sum(history, ()))
-    s.append(input)
-    inp = ' '.join(s)
-    output = openai_create(inp)
+    output = openai_create(input)
     history.append((input, output))
-    return history, history
+    return history, history, ""
 
+def clear(input,output):
+    history = []
+    input = ''
+    output = ''
+    return "","",""
 
 block = gr.Blocks()
 
+with gr.Blocks() as block:
+    gr.Markdown("""<h1><center>ChatGPT Bot</center></h1>""")
+    
+    with gr.Tab("聊天"):
+        chatbot = gr.Chatbot()
+        text_input  = gr.Textbox(placeholder=prompt_txt)
+        state = gr.State()
+        with gr.Row():
+            text_button = gr.Button("发送")
+            clear_button = gr.Button("清空")
+        
+    with gr.Tab("图像"):
+        image_output = gr.Image(width=512, height=512)
+        image_input  = gr.Textbox(placeholder=prompt_img)
+        image_size = gr.Radio(["256x256", "512x512", "1024x1024"],value="256x256",label="图片尺寸", info="选择生成的图片尺寸")
+        state = gr.State()
+        image_button = gr.Button("生成")
+        
+    text_button.click(chatgpt_chat, inputs=[text_input, state], outputs=[chatbot, state, text_input])
+    clear_button.click(clear, inputs=[text_input,state], outputs=[chatbot,state,text_input])
+    image_button.click(chatgpt_image, inputs=[image_input, image_size, state], outputs=[image_output, state, image_input])
 
-with block:
-    gr.Markdown("""<h1><center>Build Yo'own ChatGPT with OpenAI API & Gradio</center></h1>
-    """)
-    chatbot = gr.Chatbot()
-    message = gr.Textbox(placeholder=prompt)
-    state = gr.State()
-    submit = gr.Button("SEND")
-    submit.click(chatgpt_clone, inputs=[message, state], outputs=[chatbot, state])
-
-block.launch(debug = True)
+block.launch(debug=True, server_name="0.0.0.0", server_port=8000, share=True)