Upload app.py with huggingface_hub

app.py

@@ -1,20 +1,22 @@
 # -*- coding: utf-8 -*-
 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())
-from groq import Groq
+import openai
 
-# Load Groq API key from environment
-groq_api_key = os.getenv("GROQ_API_KEY")
-if not groq_api_key:
-    groq_api_key = ""
-
-client = Groq(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):
     """
-    Sends the user code to Groq LLM 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."
     messages = [
@@ -22,13 +24,13 @@ def improve_code(user_code, language):
         {"role": "user", "content": user_code}
     ]
     try:
-        response = client.chat.completions.create(
-            model="llama3-8b-8192",
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
             messages=messages,
             temperature=0.2,
             max_tokens=1024,
-            top_p=1,
-            stream=False,
+            n=1,
+            stop=None,
         )
         improved = response.choices[0].message.content.strip()
         return improved