Update app.py with api calls to OpenAI

app.py

@@ -1,7 +1,39 @@
+from transformers import pipeline
 import gradio as gr
+import openai
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Function to call OpenAI API
+def get_openai_response(api_key, model, prompt):
+    openai.api_key = api_key
+    response = openai.ChatCompletion.create(
+        model=model,
+        messages=[
+            {"role": "system", "content": "You are a helpful assistant."},
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return response['choices'][0]['message']['content']
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def generate_responses(api_key, prompt):
+    response_gpt3 = get_openai_response(api_key, "gpt-3.5-turbo", prompt)
+    response_gpt4 = get_openai_response(api_key, "gpt-4o", prompt)
+    return response_gpt3, response_gpt4
+
+# Gradio Interface
+inputs = [
+    gr.inputs.Textbox(label="OpenAI API Key"),
+    gr.inputs.Textbox(label="Input")
+]
+
+outputs = [
+    gr.outputs.Textbox(label="GPT-3.5-Turbo Output"),
+    gr.outputs.Textbox(label="GPT-4o Output")
+]
+
+gr.Interface(
+    fn=generate_responses,
+    inputs=inputs,
+    outputs=outputs,
+    title="Compare GPT Models Challenge",
+    description="Enter your OpenAI API Key and your question to get responses from GPT-3.5-Turbo and GPT-4o, compare their results."
+).launch()