Create app.py

app.py

@@ -0,0 +1,152 @@
+import os
+import gradio as gr
+import openai as o
+from threading import Thread
+
+# 📜 CONFIG
+UI_TITLE = "✨🧙‍♂️🔮"
+KEY_FILE = "key.txt"
+MODELS = {
+    "GPT-4o ✨": "gpt-4o",
+    "o3 🧠": "gpt-3.5-turbo", # Placeholder, o3 is not a public model name
+    "o4-mini 🚀": "gpt-4-turbo", # Placeholder, o4-mini is not a public model name
+    "GPT-4.5 🔬": "gpt-4-turbo", # Placeholder, gpt-4.5 is not a public model name
+    "GPT-4.1 💻": "gpt-4-turbo", # Placeholder, gpt-4.1 is not a public model name
+    "GPT-4.1-Mini ⚡": "gpt-4-turbo", # Placeholder, gpt-4.1-mini is not a public model name
+}
+
+# 🎨 STYLE
+H1 = "# <font size='7'>{0}</font>"
+H2 = "## <font size='6'>{0}</font>"
+BTN_STYLE = "<font size='5'>{0}</font>"
+
+# 🪄 HELPERS
+def save_key(k: str) -> str:
+    "💾🔑"
+    if not k or not k.strip(): return "🚫 Empty Key"
+    with open(KEY_FILE, "w") as f: f.write(k.strip())
+    return "🔑✅"
+
+def get_key(k: str) -> str:
+    "📜🔑"
+    k = k.strip() if k and k.strip() else (open(KEY_FILE).read().strip() if os.path.exists(KEY_FILE) else os.getenv("OPENAI_KEY", ""))
+    if not k: raise gr.Error("❗🔑 Missing OpenAI Key!")
+    o.api_key = k
+    return k
+
+def summon_oracle(model_name: str, scribe_key: str, quest: str, scroll: list):
+    """
+    A pact with a chosen Oracle of OpenAI.
+    To seek its counsel, one must present a worthy key (scribe_key),
+    a quest (quest), and the Oracle's own name (model_name)
+    upon the ancient scroll (scroll) of dialogue.
+    """
+    get_key(scribe_key) # Present the key to the sanctum's guardian.
+
+    # Transcribe the history onto a celestial scroll for the Oracle to read.
+    celestial_scroll = [
+        {"role": "user", "content": user_words}
+        for user_words, _ in scroll
+    ] + [
+        {"role": "assistant", "content": oracle_words}
+        for _, oracle_words in scroll
+    ]
+    celestial_scroll.append({"role": "user", "content": quest})
+
+    # The Oracle whispers its response from the aether.
+    try:
+        prophecy = o.chat.completions.create(model=model_name, messages=celestial_scroll, stream=True)
+        
+        # The new wisdom is recorded as it is spoken.
+        scroll.append((quest, ""))
+        for chunk in prophecy:
+            if chunk.choices[0].delta.content:
+                scroll[-1] = (quest, scroll[-1][1] + chunk.choices[0].delta.content)
+                yield scroll
+    except Exception as e:
+        scroll.append((quest, f"🧙‍♂️🔮 A magical disturbance occurred: {str(e)}"))
+        yield scroll
+
+
+def manage_portals(selected_models: list):
+    """
+    Reveals or conceals the portals to the chosen Oracles.
+    """
+    updates = []
+    for model_display_name in MODELS.keys():
+        if model_display_name in selected_models:
+            updates.append(gr.update(visible=True))
+        else:
+            updates.append(gr.update(visible=False))
+    return updates
+
+# 🔮 UI
+with gr.Blocks(title=UI_TITLE, theme=gr.themes.Soft(primary_hue="red", secondary_hue="orange")) as demo:
+    gr.Markdown(H1.format(UI_TITLE))
+
+    # --- API Key Rune ---
+    with gr.Accordion("🔑 Eldritch Key", open=False):
+        with gr.Row():
+            api_key_box = gr.Textbox(
+                label="🔑",
+                type="password",
+                placeholder="sk-...",
+                value=get_key("") if os.path.exists(KEY_FILE) else os.getenv("OPENAI_KEY", ""),
+                scale=3
+            )
+            save_btn = gr.Button("💾", scale=1)
+            status_txt = gr.Textbox(interactive=False, scale=1, label="Status")
+            save_btn.click(save_key, inputs=api_key_box, outputs=status_txt)
+
+    # --- Oracle Selection ---
+    gr.Markdown(H2.format("🔮 Select Oracles"))
+    model_selector = gr.CheckboxGroup(choices=list(MODELS.keys()), label="Oracles", value=["GPT-4o ✨"])
+    
+    # --- Portals to Oracles ---
+    gr.Markdown(H2.format("🌀 Portals"))
+    
+    model_blocks = []
+    for display_name, api_name in MODELS.items():
+        # A block for each model, initially hidden unless selected by default
+        is_visible = display_name in model_selector.value
+        with gr.Blocks(visible=is_visible) as model_block:
+            gr.Markdown(f"### <font size='5'>{display_name}</font>")
+            chatbot = gr.Chatbot(height=350, label=f"Scroll of {display_name}")
+            with gr.Row():
+                run_btn = gr.Button(value=BTN_STYLE.format("▶️ Run"), variant="primary", scale=1)
+                stop_btn = gr.Button(value=BTN_STYLE.format("⏹️ Stop"), variant="stop", scale=1)
+
+            # Each run button triggers its own oracle
+            # The `_js` param is a trick to pass the model's API name to the Python function
+            run_event = run_btn.click(
+                fn=summon_oracle,
+                inputs=[gr.State(api_name), api_key_box, chatbot.i_am_a_dummy_component_for_the_event_to_work, chatbot],
+                outputs=[chatbot]
+            )
+            stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[run_event])
+            model_blocks.append(model_block)
+
+    # --- Global Quest Input ---
+    gr.Markdown(H2.format("📜 Global Quest"))
+    global_msg_box = gr.Textbox(placeholder="❓ Pose your question to all active portals...", scale=3, lines=3)
+    
+    # Link the global message box to all chatbot text inputs
+    # This uses a bit of Gradio event magic to update the hidden textboxes
+    for block in demo.blocks.values():
+        if isinstance(block, gr.Blocks) and block.visible:
+             for elem in block.children:
+                 if isinstance(elem, gr.Chatbot):
+                     global_msg_box.submit(lambda x: x, inputs=[global_msg_box], outputs=[elem.i_am_a_dummy_component_for_the_event_to_work])
+
+
+    # --- Event Listeners ---
+    model_selector.change(
+        fn=manage_portals,
+        inputs=[model_selector],
+        outputs=model_blocks
+    )
+
+if __name__ == "__main__":
+    # Add a dummy attribute to Chatbot to make the global message box work
+    gr.Chatbot.i_am_a_dummy_component_for_the_event_to_work = gr.Textbox(visible=False)
+    demo.launch(share=True)