Upload app.py

app.py

@@ -1,11 +1,21 @@
+
+import os
 from openai import OpenAI
 import gradio as gr
 
+API_KEY = os.environ['API_KEY']
 messages = [
     {"role": "system", "content": "You are AI specialized in Prompt Engineering"},
 ]
-client = OpenAI(api_key="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", base_url="https://openrouter.ai/api/v1")
-
+client = OpenAI(
+    base_url="https://openrouter.ai/api/v1",
+    api_key="API_KEY",  
+    default_headers={
+        "Authorization": f"Bearer {API_KEY}",
+        "HTTP-Referer": "",  # Optional: Your app's URL
+        "X-Title": ""  # Optional: Your app name
+    }
+)
 def chatbot(input, prompt_type):
     if input:
         if prompt_type == "Zero Shot Prompt":
@@ -58,9 +68,9 @@ prompt_types = [
     "Multi-Modal Prompt",
     "Augmented Prompt"
 ]
-with gr.Blocks(theme=gr.themes.Default(primary_hue="sky")) as demo:
-    dropdown = gr.Dropdown(choices=prompt_types, label="Select Prompt Type", type="value")
-    inputs = gr.Textbox(lines=7, label="Prompt : Try - What is the value of pi")
+with gr.Blocks(theme=gr.themes.Default(primary_hue="sky")) as promptengineering:
+    dropdown = gr.Dropdown(choices=prompt_types, label="Select Prompt Type:", type="value")
+    inputs = gr.Textbox(lines=7, placeholder="Prompt : Try - What is the value of pi")
     outputs = gr.Textbox(label="Result")
-    gr.Interface(fn=chatbot, inputs=[inputs, dropdown], outputs=outputs)    
-demo.launch(share=True)
+    gr.Interface(fn=chatbot, inputs=[dropdown,inputs], outputs=outputs)    
+promptengineering.launch(share=True)