Upload app.py with huggingface_hub

app.py

@@ -6,26 +6,29 @@ import openai
 # Load OpenAI API key from environment
 openai_api_key = os.getenv("OPENAI_API_KEY")
 if not openai_api_key:
-    openai_api_key = ""
-openai.api_key = openai_api_key
+    raise ValueError("OPENAI_API_KEY environment variable not set")
+
+# Initialise OpenAI client (v1 API)
+client = openai.OpenAI(api_key=openai_api_key)
 
 def improve_code(user_code, language):
     """
     Sends the user code to OpenAI GPT model and asks for an improved version.
     """
-    system_prompt = f"You are an expert {language} developer. Refactor and improve the following code. Keep the same functionality but make it cleaner, more efficient and well commented."
+    system_prompt = (
+        f"You are an expert {language} developer. Refactor and improve the following code. "
+        "Keep the same functionality but make it cleaner, more efficient and well commented."
+    )
     messages = [
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": user_code}
     ]
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=messages,
             temperature=0.2,
             max_tokens=1024,
-            n=1,
-            stop=None,
         )
         improved = response.choices[0].message.content.strip()
         return improved
@@ -35,11 +38,19 @@ def improve_code(user_code, language):
 # Build Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Code Improver")
-    gr.Markdown("Enter your code and select the programming language. The model will return a cleaner version.")
+    gr.Markdown(
+        "Enter your code and select the programming language. The model will return a cleaner version."
+    )
     with gr.Row():
-        code_input = gr.Textbox(label="Original Code", lines=15, placeholder="Paste your code here")
+        code_input = gr.Textbox(
+            label="Original Code",
+            lines=15,
+            placeholder="Paste your code here"
+        )
         language_dropdown = gr.Dropdown(
-            choices=["Python", "JavaScript", "Java", "C++", "C#", "Go", "Ruby", "PHP"],
+            choices=[
+                "Python", "JavaScript", "Java", "C++", "C#", "Go", "Ruby", "PHP"
+            ],
             value="Python",
             label="Language"
         )