Upload app.py with huggingface_hub

app.py

@@ -19,4 +19,86 @@ openai_client = None
 if groq_api_key:
     try:
         from groq import Groq
-       
+        groq_client = Groq(api_key=groq_api_key)
+    except Exception:
+        groq_client = None
+
+if openai_api_key:
+    try:
+        import openai
+        openai.api_key = openai_api_key
+        openai_client = openai
+    except Exception:
+        openai_client = None
+
+# ----------------------------------------------------------------------
+# Helper function to improve code
+# ----------------------------------------------------------------------
+def improve_code(original_code: str) -> str:
+    """
+    Send the original code to the LLM and return the improved version.
+    """
+    if not original_code.strip():
+        return "Please provide some code to improve."
+
+    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"
+        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."
+
+# ----------------------------------------------------------------------
+# Gradio interface
+# ----------------------------------------------------------------------
+with gr.Blocks() as demo:
+    gr.Markdown("# Code Improver")
+    with gr.Row():
+        with gr.Column():
+            code_input = gr.Textbox(
+                label="Original Code",
+                placeholder="Paste your code here",
+                lines=20,
+                max_lines=200,
+                show_copy_button=True,
+            )
+        with gr.Column():
+            code_output = gr.Textbox(
+                label="Improved Code",
+                placeholder="Improved code will appear here",
+                lines=20,
+                max_lines=200,
+                show_copy_button=True,
+            )
+    improve_btn = gr.Button("Improve")
+    improve_btn.click(fn=improve_code, inputs=code_input, outputs=code_output)
+
+demo.launch()