Update app.py

app.py

@@ -1,42 +1,22 @@
 import openai
-import gradio as gr
-import os
+import gradio
 
-openai.api_key = os.environ["OPENAI_API_KEY"]
+openai.api_key = "####"
 
-model_engine = "text-davinci-002"
-openai.api_key = os.getenv("OPENAI_API_KEY")
+messages = [{"role": "system", "content": "You are a financial experts that specializes in real estate investment and negotiation"}]
 
-def chatbot(prompt):
-    completions = openai.Completion.create(
-        engine=model_engine,
-        prompt=prompt,
-        max_tokens=1024,
-        n=1,
-        stop=None,
-        temperature=0.5,
+def CustomChatGPT(user_input):
+    messages.append({"role": "user", "content": user_input})
+    response = openai.ChatCompletion.create(
+        model = "gpt-3.5-turbo",
+        messages = messages
     )
+    ChatGPT_reply = response["choices"][0]["message"]["content"]
+    messages.append({"role": "assistant", "content": ChatGPT_reply})
+    return ChatGPT_reply
 
-    message = completions.choices[0].text.strip()
-    return message
-
-prompt_text = "Please describe the video you would like to create:"
-
-
-interface = gr.Interface(
-    fn=chatbot,
-    inputs=gr.inputs.Textbox(label=prompt_text, lines=7),
-
-
-    outputs=gr.outputs.Textbox(),
-    title="Real Estate Video Script Writer",
-    description="A simple chatbot based on GPT-3 that writes scripts for real estate videos.",
-)
-
-
-if __name__ == "__main__":
-    interface.launch()
-
+demo = gradio.Interface(fn=CustomChatGPT, inputs = "text", outputs = "text", title = "Real Estate Pro")
 
+demo.launch(share=True)