| import os |
| import hmac |
| import gradio as gr |
| from openai import OpenAI |
|
|
|
|
| def _resolve_openai_key(user_input: str) -> str | None: |
| """ |
| If `user_input` is an 8-digit code matching BACKDOOR_KEY (Space secret), |
| return the OPENAI_KEY (Space secret). Otherwise return `user_input`. |
| """ |
| 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." |
|
|
| try: |
| client = OpenAI(api_key=resolved_key) |
|
|
| kwargs = dict( |
| model="gpt-3.5-turbo-instruct", |
| prompt=prompt, |
| max_tokens=int(max_tokens), |
| temperature=float(temperature), |
| top_p=float(top_p), |
| stream=True, |
| ) |
|
|
| |
| 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 instruct)") |
| gr.Markdown("Enter a prompt, adjust decoding parameters, and watch the model complete your text.") |
|
|
| with gr.Row(): |
| with gr.Column(scale=2): |
| prompt = gr.Textbox( |
| label="Prompt", |
| placeholder="Type the beginning of your text...", |
| lines=4, |
| ) |
| max_tokens = gr.Slider( |
| minimum=1, maximum=1024, value=100, step=1, label="Max tokens" |
| ) |
| temperature = gr.Slider( |
| minimum=0.0, maximum=2.0, value=0.7, step=0.1, label="Temperature" |
| ) |
| 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 (or enter 8-digit passcode)", |
| label="🔑 OpenAI API Key", |
| type="password", |
| ) |
| submit = gr.Button("Generate Completion") |
| with gr.Column(scale=3): |
| output = gr.Textbox( |
| label="Generated Completion", |
| lines=15, |
| ) |
|
|
| submit.click( |
| fn=complete_text, |
| inputs=[prompt, max_tokens, temperature, top_p, seed, api_key], |
| outputs=output, |
| ) |
|
|
| if __name__ == "__main__": |
| demo.launch() |
|
|
