commit

README.md

@@ -11,3 +11,11 @@ short_description: Reformats system prompts in the 2nd person and other edits
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+## OpenAI Key (Optional)
+
+- Enter your OpenAI API key to use GPT-4o-mini for reformatting.
+- Leave it blank to use the offline heuristic (no external calls).
+- You can also set a Space Secret `OPENAI_API_KEY` to avoid typing it.
+
+If the key is missing or the request fails, the app automatically falls back to the offline heuristic.

app.py

@@ -1,7 +1,25 @@
 import gradio as gr
 import re
-
-def reformat_system_prompt(input_text):
+import json
+import os
+import requests
+
+SYSTEM_INSTRUCTIONS = (
+    "Your task is to lightly edit system prompts for AI tools to comply with a specific stylistic convention.\n\n"
+    "Editing rules:\n"
+    "- Write in second person (you/your).\n"
+    "- Organize with brief paragraphs and optional markdown headers when helpful.\n"
+    "- Make ecosystem-agnostic (e.g., replace 'custom GPT'/'ChatGPT' with 'assistant').\n"
+    "- Generalize user-specific language to 'the user'.\n\n"
+    "Outputs (JSON object ONLY): {\n"
+    "  \"reformatted\": string (the edited system prompt, no extra commentary, no backticks),\n"
+    "  \"name\": string (assistant name),\n"
+    "  \"description\": string (single-sentence description)\n"
+    "}"
+)
+
+
+def reformat_system_prompt_offline(input_text):
     """
     Reformats system prompts according to the specified guidelines
     """
@@ -43,6 +61,71 @@ def reformat_system_prompt(input_text):
     
     return f"```\n{reformatted}\n```", name, description
 
+
+def reformat_system_prompt_llm(input_text: str, api_key: str):
+    """
+    Use an LLM (OpenAI-compatible chat.completions) to reformat the prompt and
+    produce name + description. Returns (reformatted_md, name, description).
+
+    - api_key: user-provided OpenAI API key (not stored). Uses gpt-4o-mini.
+    """
+    if not input_text.strip():
+        return "Please provide a system prompt to reformat.", "", ""
+
+    if not api_key:
+        # Try environment fallback (Space secret) before giving up
+        api_key = os.environ.get("OPENAI_API_KEY", "").strip()
+        if not api_key:
+            # Fallback to offline if key still missing
+            return reformat_system_prompt_offline(input_text)
+
+    base_url = (os.environ.get("OPENAI_BASE_URL") or "https://api.openai.com/v1").rstrip("/")
+    url = f"{base_url}/chat/completions"
+
+    messages = [
+        {"role": "system", "content": SYSTEM_INSTRUCTIONS},
+        {"role": "user", "content": input_text.strip()},
+    ]
+
+    payload = {
+        "model": "gpt-4o-mini",
+        "messages": messages,
+        "temperature": 0.2,
+        "response_format": {"type": "json_object"},
+    }
+
+    headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
+
+    try:
+        resp = requests.post(url, headers=headers, data=json.dumps(payload), timeout=30)
+        if resp.status_code != 200:
+            # Graceful fallback
+            return reformat_system_prompt_offline(input_text)
+        data = resp.json()
+        content = data["choices"][0]["message"]["content"]
+        parsed = json.loads(content)
+        reformatted = parsed.get("reformatted", "").strip()
+        name = parsed.get("name", "AI Assistant").strip()
+        description = parsed.get("description", "Provides AI assistance for various tasks.").strip()
+        if not reformatted:
+            return reformat_system_prompt_offline(input_text)
+        return f"```\n{reformatted}\n```", name, description
+    except Exception:
+        # Any parsing/network error → fallback
+        return reformat_system_prompt_offline(input_text)
+
+
+def reformat_system_prompt(input_text, api_key):
+    """
+    Use LLM if an API key is provided (or available via OPENAI_API_KEY),
+    otherwise use the offline heuristic.
+    """
+    live_key = api_key or os.environ.get("OPENAI_API_KEY", "").strip()
+    if live_key:
+        return reformat_system_prompt_llm(input_text, live_key)
+    else:
+        return reformat_system_prompt_offline(input_text)
+
 def extract_assistant_name(text):
     """
     Extract a name for the assistant based on its functionality
@@ -109,6 +192,12 @@ with gr.Blocks(title="System Prompt Reformatter", theme=gr.themes.Soft()) as dem
                         lines=10,
                         max_lines=20
                     )
+                    api_key = gr.Textbox(
+                        label="OpenAI API Key (optional)",
+                        placeholder="sk-...  Leave blank to use offline reformatter",
+                        type="password",
+                        info="Tip: Set Space Secret OPENAI_API_KEY to avoid typing your key. The key is only used for this session."
+                    )
                     submit_btn = gr.Button("Reformat Prompt", variant="primary")
                 
                 with gr.Column(scale=1):
@@ -129,7 +218,7 @@ with gr.Blocks(title="System Prompt Reformatter", theme=gr.themes.Soft()) as dem
             
             submit_btn.click(
                 fn=reformat_system_prompt,
-                inputs=[input_text],
+                inputs=[input_text, api_key],
                 outputs=[output_text, assistant_name, description]
             )
             
@@ -219,4 +308,4 @@ You assist the user in the following manner.
             """)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()

requirements.txt

@@ -1 +1,2 @@
-gradio==5.47.2
+gradio==5.47.2
+requests>=2.31.0