Update app.py

app.py

@@ -10,131 +10,30 @@ openai.api_key = os.environ['OPENAI_API_KEY']
 
 user_db = {os.environ['username1']: os.environ['password1'], os.environ['username2']: os.environ['password2'], os.environ['username3']: os.environ['password3']}
 
-messages = [{"role": "system", "content": 'You are a helpful assistant.'}]
 
 
 
-def roleChoice(role):
-    global messages
-    messages = [{"role": "system", "content": role}]
-    return "role:" + role
-
-
-def audioGPT(audio):
-    global messages 
-
-    audio_file = open(audio, "rb")
-    transcript = openai.Audio.transcribe("whisper-1", audio_file)
-
-    messages.append({"role": "user", "content": transcript["text"]})
-
-    response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
-
-    system_message = response["choices"][0]["message"]
-    messages.append(system_message)
-
-    chats = ""
-    for msg in messages:
-        if msg['role'] != 'system':
-            chats += msg['role'] + ": " + msg['content'] + "\n\n"
-
-    return chats
-
-
 def textGPT(text):
-    global messages
-
-    messages.append({"role": "user", "content": text})
-
-    response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
-
-    system_message = response["choices"][0]["message"]
-    messages.append(system_message)
-
-    chats = ""
-    for msg in messages:
-        if msg['role'] != 'system':
-            chats += msg['role'] + ": " + msg['content'] + "\n\n"
-
-    return chats
-
-
-def siriGPT(audio):
-    global messages 
-
-    audio_file = open(audio, "rb")
-    transcript = openai.Audio.transcribe("whisper-1", audio_file)
-
-    messages.append({"role": "user", "content": transcript["text"]})
-
-    response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
-
-    system_message = response["choices"][0]["message"]
-    messages.append(system_message)
-
-    lang = detect(system_message['content'])
+    messages = [{"role": "system", "content": 'You are a coding assistant.'}]
 
-    narrate_ans = gTTS(text=system_message['content'], lang=lang, slow=False) 
-    narrate_ans.save("narrate.wav") 
+    cuda_codes = "Translate this CUDA code into HIP code:\n" + text + "\n\n###"
 
-    return "narrate.wav"
-    
+    messages.append({"role": "user", "content": cuda_codes})
 
-def fileGPT(prompt, file_obj):
-    global messages 
-
-    file_text = extract_text(file_obj.name)
-    text = prompt + "\n\n" + file_text
-    
-    messages.append({"role": "user", "content": text})
-    
-    response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
+    response = openai.ChatCompletion.create(model="davinci:ft-zhaoyi-2023-06-21-07-18-01", messages=messages, stop="###")
 
     system_message = response["choices"][0]["message"]
-    messages.append(system_message)
-
-    chats = ""
-    for msg in messages:
-        if msg['role'] != 'system':
-            chats += msg['role'] + ": " + msg['content'] + "\n\n"
+    hip_codes = system_message["content"]
 
     return chats
 
 
 
-def clear():
-    global messages
-    messages = [{"role": "system", "content": 'You are a helpful technology assistant.'}]
-    return
-    
-def show():
-    global messages
-    chats = ""
-    for msg in messages:
-        if msg['role'] != 'system':
-            chats += msg['role'] + ": " + msg['content'] + "\n\n"
-
-    return chats
-
 
-with gr.Blocks() as chatHistory:
-    gr.Markdown("Click the Clear button below to remove all the chat history.")
-    clear_btn = gr.Button("Clear")
-    clear_btn.click(fn=clear, inputs=None, outputs=None, queue=False)
 
-    gr.Markdown("Click the Display button below to show all the chat history.")
-    show_out = gr.Textbox()
-    show_btn = gr.Button("Display")
-    show_btn.click(fn=show, inputs=None, outputs=show_out, queue=False)
-
-
-
-role = gr.Interface(fn=roleChoice, inputs="text", outputs="text", description = "Choose your GPT roles, e.g. You are a helpful technology assistant. 你是一位 IT 架构师。 你是一位开发者关系顾问。你是一位机器学习工程师。你是一位高级 C++ 开发人员 ")
 text = gr.Interface(fn=textGPT, inputs="text", outputs="text")
-audio = gr.Interface(fn=audioGPT, inputs=gr.Audio(source="microphone", type="filepath"), outputs="text")
-siri = gr.Interface(fn=siriGPT, inputs=gr.Audio(source="microphone", type="filepath"), outputs = "audio")
-file = gr.Interface(fn=fileGPT, inputs=["text", "file"], outputs="text", description = "Enter prompt sentences and your PDF. e.g. lets think step by step, summarize this following text:  或者 让我们一步一步地思考，总结以下的内容:")
-demo = gr.TabbedInterface([role, text, audio, siri, file, chatHistory], [ "roleChoice", "chatGPT", "audioGPT", "siriGPT", "fileGPT", "ChatHistory"])
+
+demo = gr.TabbedInterface([text], [ "HipifyPlus"])
 
 if __name__ == "__main__":
     demo.launch(enable_queue=False, auth=lambda u, p: user_db.get(u) == p,