Upload app.py with huggingface_hub

app.py

@@ -6,17 +6,17 @@ from starlette.exceptions import StarletteDeprecationWarning
 warnings.filterwarnings("ignore", category=StarletteDeprecationWarning)
 import asyncio
 asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
-import openai
+import groq
 
-# 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
+# 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
 
 def improve_code(user_code, language):
     """
-    Sends the user code to OpenAI GPT model and asks for an improved version.
+    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."
     messages = [
@@ -24,8 +24,8 @@ def improve_code(user_code, language):
         {"role": "user", "content": user_code}
     ]
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
+        response = groq.ChatCompletion.create(
+            model="mixtral-8x7b-32768",
             messages=messages,
             temperature=0.2,
             max_tokens=1024,