Upload app.py with huggingface_hub

app.py

@@ -4,77 +4,57 @@ import warnings
 
 import gradio as gr
 
-# Suppress Starlette deprecation warnings that appear in Spaces
+# Suppress warnings that can appear in the Space environment
 warnings.filterwarnings("ignore", category=UserWarning)
 
 # ----------------------------------------------------------------------
-# LLM client setup
+# Groq client setup (only)
 # ----------------------------------------------------------------------
 groq_api_key = os.getenv("GROQ_API_KEY")
-openai_api_key = os.getenv("OPENAI_API_KEY")
-
 groq_client = None
-openai_client = None
 
 if groq_api_key:
     try:
         from groq import Groq
         groq_client = Groq(api_key=groq_api_key)
-    except Exception:
+    except Exception as e:
+        # If the library cannot be imported or instantiated, keep client as None
         groq_client = None
-
-if openai_api_key:
-    try:
-        import openai
-        openai.api_key = openai_api_key
-        openai_client = openai
-    except Exception:
-        openai_client = None
+        print(f"Failed to initialise Groq client: {e}")
 
 # ----------------------------------------------------------------------
-# Helper function to improve code
+# Helper function to improve code using Groq
 # ----------------------------------------------------------------------
 def improve_code(original_code: str) -> str:
     """
-    Send the original code to the LLM and return the improved version.
+    Sends the original code to the Groq LLM and returns an improved version.
+    If no API key is configured, returns an informative message.
     """
     if not original_code.strip():
         return "Please provide some code to improve."
 
+    if not groq_client:
+        return "No Groq API key configured. Set the GROQ_API_KEY environment variable."
+
     prompt = (
-        "You are a senior software engineer. "
-        "Improve the following code for readability, efficiency, and best practices. "
-        "Return only the improved code without any explanation."
-        "\n\nOriginal code:\n"
+        "You are a senior software engineer. Improve the following code for readability, "
+        "efficiency, and best practices. Return only the improved code without any explanation.\n\n"
+        "Original code:\n"
         f"{original_code}"
     )
 
-    # Prefer Groq if available, otherwise fallback to OpenAI
-    if groq_client:
-        try:
-            response = groq_client.chat.completions.create(
-                model="llama3-70b-8192",
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.2,
-                max_tokens=2000,
-            )
-            return response.choices[0].message.content.strip()
-        except Exception as e:
-            return f"Error with Groq: {e}"
-
-    if openai_client:
-        try:
-            response = openai_client.ChatCompletion.create(
-                model="gpt-4o-mini",
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.2,
-                max_tokens=2000,
-            )
-            return response.choices[0].message.content.strip()
-        except Exception as e:
-            return f"Error with OpenAI: {e}"
-
-    return "No LLM API key configured. Please set GROQ_API_KEY or OPENAI_API_KEY."
+    try:
+        response = groq_client.chat.completions.create(
+            model="llama3-70b-8192",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.2,
+            max_tokens=2000,
+        )
+        # The content may be None if the API fails; guard against that
+        improved = response.choices[0].message.content
+        return improved.strip() if improved else "No content returned from Groq."
+    except Exception as e:
+        return f"Error communicating with Groq: {e}"
 
 # ----------------------------------------------------------------------
 # Gradio interface