Update app.py

app.py

@@ -1,20 +1,24 @@
 import gradio as gr
 import openai
+import os
 
-# Initialize the OpenAI API with your API key
-openai.api_key = "openai-api-key"
+# Initialize the OpenAI API with your API key from environment variables
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
 def generate_text(prompt, max_length=100, temperature=1.0):
-    # Generate text using GPT-3
-    response = openai.Completion.create(
-        engine="text-davinci-003",  # You can also use other engines like "gpt-3.5-turbo"
-        prompt=prompt,
-        max_tokens=max_length,
-        temperature=temperature,
-        n=1,
-        stop=None
-    )
-    return response.choices[0].text.strip()
+    try:
+        # Generate text using OpenAI's GPT-3 (text-davinci-003)
+        response = openai.Completion.create(
+            engine="text-davinci-003",  # Specify the engine
+            prompt=prompt,
+            max_tokens=max_length,
+            temperature=temperature,
+            n=1,
+            stop=None
+        )
+        return response.choices[0].text.strip()
+    except openai.error.OpenAIError as e:
+        return f"An error occurred: {str(e)}"
 
 # Create a Gradio interface
 iface = gr.Interface(
@@ -25,9 +29,9 @@ iface = gr.Interface(
         gr.Slider(minimum=0.1, maximum=1.5, value=1.0, label="Temperature"),
     ],
     outputs=gr.Textbox(label="Generated Text"),
-    title="GPT Text Generator",
-    description="Enter a prompt and generate text using the GPT model."
+    title="GP Text Generator",
+    description="Enter a prompt and generate text using OpenAI's GPT (text-davinci-003)."
 )
 
 # Launch the interface
-iface.launch(share=True)
+iface.launch(share=True)