Update app.py

app.py

@@ -2,43 +2,47 @@ import gradio as gr
 import requests
 import os
 
-# Load your API key from environment variable
+# Read API key from environment variable
 API_KEY = os.getenv("OPENAI_API_KEY")
 
-# List of supported programming languages
 languages = ["Python", "C", "C++", "Java", "JavaScript", "Solidity", "SQL", "Bash"]
 
 def code_assistant(prompt, language="Python"):
-    """
-    Sends prompt to OpenAI API (or Hugging Face endpoint) with system instructions
-    to act as an expert programmer in the selected language.
-    """
-    system_prompt = f"You are an expert {language} programmer. Only respond with code or explanation as requested. No extra commentary."
-    
-    response = requests.post(
-        "https://api.openai.com/v1/chat/completions",
-        headers={"Authorization": f"Bearer {API_KEY}"},
-        json={
-            "model": "gpt-4o-mini",
-            "messages": [
-                {"role": "system", "content": system_prompt},
-                {"role": "user", "content": prompt}
-            ]
-        }
-    )
+    # Check if API key exists
+    if not API_KEY:
+        return "Error: API key not set. Please add your OPENAI_API_KEY in Space Secrets and restart the Space."
+
+    system_prompt = f"You are an expert {language} programmer. Only respond with code or explanation as requested. Wrap code in Markdown-style code blocks."
     
-    # Extract the output text
     try:
-        return response.json()["choices"][0]["message"]["content"]
+        response = requests.post(
+            "https://api.openai.com/v1/chat/completions",
+            headers={"Authorization": f"Bearer {API_KEY}"},
+            json={
+                "model": "gpt-4o-mini",
+                "messages": [
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": prompt}
+                ]
+            }
+        )
+
+        text = response.json()["choices"][0]["message"]["content"]
+
+        # Ensure output is wrapped in Markdown code block
+        if not text.startswith("```"):
+            text = f"```{language.lower()}\n{text}\n```"
+        return text
+
     except Exception as e:
-        return f"Error: {e}\nResponse: {response.text}"
+        return f"Error: {e}\nResponse: {response.text if 'response' in locals() else 'No response'}"
 
-# Gradio interface
+# Gradio UI
 iface = gr.Interface(
     fn=code_assistant,
     inputs=[gr.Textbox(label="Your Prompt", placeholder="Write, explain, or debug code..."),
             gr.Dropdown(languages, label="Language")],
-    outputs=gr.Textbox(label="Output"),
+    outputs=gr.Textbox(label="Output", lines=20, interactive=False),
     title="TheWizard 💬 — Code Assistant",
     description="Ask for code, explanations, or debugging in multiple languages. Focused only on coding."
 )