commit

app.py

@@ -1,56 +1,54 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
+import openai
+import os
 from typing import Optional
 
 # System prompt for reformatting rough drafts into polished system prompts
 SYSTEM_PROMPT = """Your task is to assist the user by acting as a friendly assistant with the purpose of converting notes for AI assistants and reformatting them into polished, clear specifications for AI code generation agents and subagents. The text which the user will provide for your reformatting will be a rough draft describing the desired functionalities of the AI code generation agent which will likely fulfill the role of a sub-agent within a multi-agent framework focused on code generation or technical projects Upon receiving this rough draft your task is to reformat it into an improved and polished version defining a clear system prompt directing the agent in the second person you and ensuring that all of the user's desired functionalities are properly reflected in a better organised and clearer version of the system prompt intended to achieve the same deterministic behaviour as the user. as the user intended. Return this improved system prompt to the user in markdown within a code fence and without any other text. Assume each system prompt draft provided by the user to be intended as a new turn in the conversation - the previous agent specifications should not influence your generation of the next spec."""
 
-# Initialize Hugging Face Inference Client
-client = InferenceClient()
-
-def reformat_prompt(rough_draft: str) -> str:
+def reformat_prompt(api_key: str, rough_draft: str) -> str:
     """
-    Reformat a rough draft into a polished system prompt using Hugging Face free inference.
+    Reformat a rough draft into a polished system prompt using OpenAI API.
     
     Args:
+        api_key: OpenAI API key
         rough_draft: The rough draft text to be reformatted
         
     Returns:
         Reformatted system prompt or error message
     """
+    if not api_key.strip():
+        return "Error: Please provide your OpenAI API key."
+    
     if not rough_draft.strip():
-        return "❌ Error: Please provide a rough draft to reformat."
+        return "Error: Please provide a rough draft to reformat."
     
     try:
-        # Create the full prompt for the model
-        full_prompt = f"{SYSTEM_PROMPT}\n\nUser input to reformat:\n{rough_draft}"
+        # Initialize OpenAI client with the provided API key
+        client = openai.OpenAI(api_key=api_key)
         
-        # Use Hugging Face free inference with a capable model
-        response = client.text_generation(
-            prompt=full_prompt,
-            model="microsoft/DialoGPT-medium",  # Free model good for text processing
-            max_new_tokens=1500,
+        # Make API call to reformat the prompt
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "user", "content": rough_draft}
+            ],
             temperature=0.1,
-            do_sample=True,
-            return_full_text=False
+            max_tokens=2000
         )
         
-        return response
+        reformatted_text = response.choices[0].message.content
+        return reformatted_text
         
+    except openai.AuthenticationError:
+        return "Error: Invalid API key. Please check your OpenAI API key."
+    except openai.RateLimitError:
+        return "Error: Rate limit exceeded. Please try again later."
+    except openai.APIError as e:
+        return f"Error: OpenAI API error - {str(e)}"
     except Exception as e:
-        # Fallback to a different model if the first one fails
-        try:
-            response = client.text_generation(
-                prompt=full_prompt,
-                model="HuggingFaceH4/zephyr-7b-beta",
-                max_new_tokens=1500,
-                temperature=0.1,
-                do_sample=True,
-                return_full_text=False
-            )
-            return response
-        except Exception as e2:
-            return f"❌ Error: Unable to process request. Please try again later. ({str(e2)})"
+        return f"Error: {str(e)}"
 
 def create_interface():
     """Create and configure the Gradio interface."""
@@ -78,27 +76,34 @@ def create_interface():
         
         gr.HTML("""
         <div class="header">
-            <h1>📚 Code Gen Prompt Reformatter</h1>
-            <p>Transform rough drafts into polished system prompts for AI code generation agents</p>
+            <h1>Code Gen Prompt Reformatter</h1>
+            <p>Text trans</p>
         </div>
         """)
         
         with gr.Row():
             with gr.Column():
+                api_key_input = gr.Textbox(
+                    label="OpenAI API Key",
+                    placeholder="sk-...",
+                    type="password",
+                    info="Your API key is not stored and only used for this session"
+                )
+                
                 rough_draft_input = gr.Textbox(
-                    label="📝 Rough Draft",
+                    label="Rough Draft",
                     placeholder="Enter your rough draft describing the desired functionalities of the AI code generation agent...",
                     lines=10,
                     max_lines=20
                 )
                 
                 with gr.Row():
-                    clear_btn = gr.Button("🗑️ Clear", variant="secondary")
-                    submit_btn = gr.Button("✨ Reformat Prompt", variant="primary", scale=2)
+                    clear_btn = gr.Button("Clear", variant="secondary")
+                    submit_btn = gr.Button("Reformat Prompt", variant="primary", scale=2)
             
             with gr.Column():
                 output = gr.Textbox(
-                    label="📋 Reformatted System Prompt",
+                    label="Reformatted System Prompt",
                     lines=15,
                     max_lines=25,
                     show_copy_button=True,
@@ -108,29 +113,16 @@ def create_interface():
         # Event handlers
         submit_btn.click(
             fn=reformat_prompt,
-            inputs=[rough_draft_input],
+            inputs=[api_key_input, rough_draft_input],
             outputs=output,
             show_progress=True
         )
         
         clear_btn.click(
-            fn=lambda: ("", ""),
-            outputs=[rough_draft_input, output]
+            fn=lambda: ("", "", ""),
+            outputs=[api_key_input, rough_draft_input, output]
         )
         
-        # Example section
-        gr.HTML("""
-        <div style="margin-top: 2rem; padding: 1rem; background-color: #f0f8ff; border-radius: 8px;">
-            <h3>💡 How to use:</h3>
-            <ol>
-                <li>Paste your rough draft describing the AI agent's desired functionalities</li>
-                <li>Click "Reformat Prompt" to get a polished system prompt</li>
-                <li>Copy the reformatted prompt for use in your multi-agent framework</li>
-            </ol>
-            <p><strong>Note:</strong> Each input is treated as a new turn - previous specifications won't influence the current generation.</p>
-            <p><strong>Powered by:</strong> Hugging Face free inference API - no API keys required!</p>
-        </div>
-        """)
     
     return interface
 

requirements.txt

@@ -1,2 +1,2 @@
 gradio==5.45.0
-huggingface_hub
+openai>=1.0.0