Update app.py

app.py

@@ -1,35 +1,65 @@
+import os
+import hmac
 import gradio as gr
 from openai import OpenAI
 
 
-def complete_text(prompt, max_tokens, temperature, top_p, openai_api_key):
+def _resolve_openai_key(user_input: str) -> str | None:
     """
-    Get a plain text completion from OpenAI gpt-3.5-turbo-instruct.
+    If `user_input` is an 8-digit code matching BACKDOOR_KEY (Space secret),
+    return the OPENAI_KEY (Space secret). Otherwise return `user_input`.
     """
-    if not openai_api_key:
-        return "⚠️ Please enter a valid OpenAI API key."
-
-    client = OpenAI(api_key=openai_api_key)
-
-    response_text = ""
-    stream = client.completions.create(
-        model="gpt-3.5-turbo-instruct",
-        prompt=prompt,
-        max_tokens=max_tokens,
-        temperature=temperature,
-        top_p=top_p,
-        stream=True,
-    )
+    user_input = (user_input or "").strip()
+
+    backdoor_code = (os.getenv("BACKDOOR_KEY") or "").strip()
+    is_candidate_code = user_input.isdigit() and len(user_input) == 8 and backdoor_code
+
+    if is_candidate_code and hmac.compare_digest(user_input, backdoor_code):
+        resolved = os.getenv("OPENAI_KEY")
+        return resolved if resolved else None
+
+    return user_input if user_input else None
+
+
+def complete_text(prompt, max_tokens, temperature, top_p, seed, openai_api_key_input):
+    """
+    Get a plain text completion from OpenAI with optional seed.
+    """
+    resolved_key = _resolve_openai_key(openai_api_key_input)
+    if not resolved_key:
+        return "⚠️ Missing or invalid API key."
 
-    for event in stream:
-        if hasattr(event, "choices") and event.choices:
-            token = event.choices[0].text or ""
-            response_text += token
-            yield response_text
+    try:
+        client = OpenAI(api_key=resolved_key)
+
+        kwargs = dict(
+            model="gpt-3.5-turbo-instruct",  # legacy instruct model
+            prompt=prompt,
+            max_tokens=int(max_tokens),
+            temperature=float(temperature),
+            top_p=float(top_p),
+            stream=True,
+        )
+
+        # Only pass seed if user entered one
+        if seed and str(seed).isdigit():
+            kwargs["seed"] = int(seed)
+
+        response_text = ""
+        stream = client.completions.create(**kwargs)
+
+        for event in stream:
+            if hasattr(event, "choices") and event.choices:
+                token = event.choices[0].text or ""
+                response_text += token
+                yield response_text
+
+    except Exception as e:
+        yield f"❌ Error: {type(e).__name__}. Check model, key, or parameters."
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("## ✍️ Text Completion Demo (OpenAI gpt-3.5-turbo-instruct)")
+    gr.Markdown("## ✍️ Text Completion Demo (OpenAI instruct)")
     gr.Markdown("Enter a prompt, adjust decoding parameters, and watch the model complete your text.")
 
     with gr.Row():
@@ -48,8 +78,13 @@ with gr.Blocks() as demo:
             top_p = gr.Slider(
                 minimum=0.1, maximum=1.0, value=1.0, step=0.05, label="Top-p"
             )
+            seed = gr.Textbox(
+                placeholder="Optional integer seed for reproducibility",
+                label="🎲 Seed",
+                type="text",
+            )
             api_key = gr.Textbox(
-                placeholder="sk-... Paste your OpenAI API key here",
+                placeholder="sk-... Paste your OpenAI API key here (or enter 8-digit passcode)",
                 label="🔑 OpenAI API Key",
                 type="password",
             )
@@ -62,11 +97,9 @@ with gr.Blocks() as demo:
 
     submit.click(
         fn=complete_text,
-        inputs=[prompt, max_tokens, temperature, top_p, api_key],
+        inputs=[prompt, max_tokens, temperature, top_p, seed, api_key],
         outputs=output,
     )
 
 if __name__ == "__main__":
     demo.launch()
-
-