Update app.py

app.py

@@ -1,54 +1,40 @@
-import os
 import openai
 import gradio as gr
+import os
 
-#if you have OpenAI API key as an environment variable, enable the below
-openai.api_key = os.getenv("OPENAI_API_KEY")
-
-
-
-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: "
-
-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:"]
-    )
-
-    return response.choices[0].text
-
-
-
-def chatgpt_clone(input, history):
-    history = history or []
-    s = list(sum(history, ()))
-    s.append(input)
-    inp = ' '.join(s)
-    output = openai_create(inp)
-    history.append((input, output))
-    return history, history
-
-
-block = gr.Blocks()
-
-
-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)
+# Set OpenAI API key from environment variable
+openai.api_key = os.getenv(OPENAI_API_KEY)
+
+# Define a function to interact with OpenAI's ChatGPT
+def chat_with_openai(input_text):
+    try:
+        response = openai.Completion.create(
+            engine="gpt-4",  # Use the model you want (e.g., 'gpt-3.5-turbo', 'gpt-4')
+            prompt=input_text,
+            max_tokens=150,
+            n=1,
+            stop=None,
+            temperature=0.7,
+        )
+        answer = response.choices[0].text.strip()
+        return answer
+    except Exception as e:
+        return str(e)
+
+# Define the Gradio interface
+with gr.Blocks() as ui:
+    gr.Markdown("# Chatbot with OpenAI API")
+
+    chatbot = gr.Chatbot(label="OpenAI Chatbot")
+    msg = gr.Textbox(label="Enter your message here:")
+    submit_btn = gr.Button("Submit")
+
+    def on_submit(message, chat_history):
+        response = chat_with_openai(message)
+        chat_history.append((message, response))
+        return chat_history, ""
+
+    submit_btn.click(on_submit, inputs=[msg, chatbot], outputs=[chatbot, msg])
+
+# Launch the Gradio app
+ui.launch()