Upload app.py with huggingface_hub

app.py

@@ -3,50 +3,69 @@ import os
 import gradio as gr
 import warnings
 from starlette.exceptions import StarletteDeprecationWarning
+
 warnings.filterwarnings("ignore", category=StarletteDeprecationWarning)
-import asyncio
-asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
+
 import groq
 
 # Load Groq API key from environment
 groq_api_key = os.getenv("GROQ_API_KEY")
 if not groq_api_key:
-    groq_api_key = ""
-groq.api_key = groq_api_key
+    raise ValueError("GROQ_API_KEY environment variable not set")
+
+# Initialise Groq client
+client = groq.Client(api_key=groq_api_key)
+
 
 def improve_code(user_code, language):
     """
     Sends the user code to Groq 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}
+        {"role": "user", "content": user_code},
     ]
+
     try:
-        response = groq.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="mixtral-8x7b-32768",
             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.")
+    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"
+            label="Language",
         )
     improve_button = gr.Button("Improve")
     output_box = gr.Textbox(label="Improved Code", lines=15)
@@ -54,7 +73,7 @@ with gr.Blocks() as demo:
     improve_button.click(
         fn=improve_code,
         inputs=[code_input, language_dropdown],
-        outputs=output_box
+        outputs=output_box,
     )
 
 demo.launch(server_name="0.0.0.0")