Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+import os
+import gradio as gr
+import openai
+
+# Load OpenAI API key from environment
+openai_api_key = os.environ.get("OPENAI_API_KEY")
+if openai_api_key is None:
+    raise ValueError("OPENAI_API_KEY environment variable not set")
+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."
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": user_code}
+    ]
+    try:
+        response = openai.ChatCompletion.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
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# 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.")
+    with gr.Row():
+        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"],
+            value="Python",
+            label="Language"
+        )
+    improve_button = gr.Button("Improve")
+    output_box = gr.Textbox(label="Improved Code", lines=15)
+
+    improve_button.click(
+        fn=improve_code,
+        inputs=[code_input, language_dropdown],
+        outputs=output_box
+    )
+
+if __name__ == "__main__":
+    demo.launch()