Upload app.py with huggingface_hub

app.py

@@ -3,69 +3,52 @@ 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 openai
 
-import groq
-
-# Load Groq API key from environment
-groq_api_key = os.getenv("GROQ_API_KEY")
-if not groq_api_key:
-    raise ValueError("GROQ_API_KEY environment variable not set")
-
-# Initialise Groq client
-client = groq.Client(api_key=groq_api_key)
-
+# 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
 
 def improve_code(user_code, language):
+    if startup_error:
+        return f"Error: {startup_error}"
     """
-    Sends the user code to Groq model and asks for an improved version.
+    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},
+        {"role": "user", "content": user_code}
     ]
-
     try:
-        response = client.chat.completions.create(
-            model="mixtral-8x7b-32768",
+        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."
-    )
+    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)
@@ -73,7 +56,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")