Update app.py

app.py

@@ -6,10 +6,10 @@ import os
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
 # Define a function to interact with OpenAI's ChatGPT
-def chat_with_openai(input_text):
+def chat_with_openai(input_text, gpt_model):
     try:
         response = openai.Completion.create(
-            engine="gpt-4o-mini",  # Use the model you want (e.g., 'gpt-3.5-turbo', 'gpt-4')
+            engine=gpt_model,  # Use the model you want (e.g., 'gpt-3.5-turbo', 'gpt-4')
             prompt=input_text,
             max_tokens=150,
             n=1,
@@ -21,20 +21,29 @@ def chat_with_openai(input_text):
     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):
+
+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])
+    
+
+# Define the Gradio interface
+with gr.Blocks(theme="Hev832/niceandsimple") as ui:
+    gr.Markdown("# ChatGPT Clone")
+
+    chatbot = gr.Chatbot(label="OpenAI Chatbot")
+    with gr.Row():
+        msg = gr.Textbox(label="Enter your message here:")
+        model gr.Dropdown(["gpt-4o-mini", "gpt-3.5-turbo", " gpt-4o"], label="your gpt model")
+    submit_btn = gr.Button("Submit")
+
+    
+
+    submit_btn.click(on_submit, inputs=[msg, model, chatbot], outputs=[chatbot, msg])
 
 # Launch the Gradio app
 ui.launch()