Upload app.py with huggingface_hub

app.py

@@ -4,11 +4,13 @@ import warnings
 
 import gradio as gr
 
-# Suppress warnings that can appear in the Space environment
+# ----------------------------------------------------------------------
+# Suppress noisy warnings in the Space environment
+# ----------------------------------------------------------------------
 warnings.filterwarnings("ignore", category=UserWarning)
 
 # ----------------------------------------------------------------------
-# Groq client setup
+# Groq client initialization (reads API key from environment)
 # ----------------------------------------------------------------------
 groq_api_key = os.getenv("GROQ_API_KEY")
 groq_client = None
@@ -18,17 +20,42 @@ if groq_api_key:
         from groq import Groq
         groq_client = Groq(api_key=groq_api_key)
     except Exception as e:
-        # If the library cannot be imported or instantiated, keep client as None
+        print(f"Failed to import or initialise Groq client: {e}")
         groq_client = None
-        print(f"Failed to initialise Groq client: {e}")
+else:
+    print("GROQ_API_KEY not set. The app will work in demo mode with placeholder messages.")
 
 # ----------------------------------------------------------------------
-# Helper function to improve code using Groq
+# Helper: list of supported languages (common set)
+# ----------------------------------------------------------------------
+SUPPORTED_LANGUAGES = [
+    "Python",
+    "JavaScript",
+    "Java",
+    "C++",
+    "C#",
+    "Go",
+    "Ruby",
+    "PHP",
+    "TypeScript",
+    "HTML",
+    "CSS",
+    "Swift",
+    "Kotlin",
+    "Rust",
+    "Scala",
+    "Perl",
+    "R",
+    "Shell",
+]
+
 # ----------------------------------------------------------------------
-def improve_code(original_code: str) -> str:
+# Function: improve existing code
+# ----------------------------------------------------------------------
+def improve_code(original_code: str, language: str) -> str:
     """
-    Sends the original code to the Groq LLM and returns an improved version.
-    If no API key is configured, returns an informative message.
+    Sends the original code to the LLM and returns an improved version.
+    The language selector is used to guide the model.
     """
     if not original_code.strip():
         return "Please provide some code to improve."
@@ -37,10 +64,10 @@ def improve_code(original_code: str) -> str:
         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\n"
-        "Original code:\n"
-        f"{original_code}"
+        f"You are a senior software engineer. Improve the following {language} code for "
+        "readability, efficiency and best practices. Return only the improved code "
+        "without any explanation.\n\n"
+        f"Original code:\n{original_code}"
     )
 
     try:
@@ -56,26 +83,102 @@ def improve_code(original_code: str) -> str:
         return f"Error communicating with Groq: {e}"
 
 # ----------------------------------------------------------------------
-# Gradio interface
+# Function: generate full web code from an idea
+# ----------------------------------------------------------------------
+def generate_web_code(idea: str, language: str) -> str:
+    """
+    Generates a complete web project (HTML/CSS/JS or a single‑file script)
+    based on the user's idea and the selected language.
+    """
+    if not idea.strip():
+        return "Please describe the web idea you want to build."
+
+    if not groq_client:
+        return "No Groq API key configured. Set the GROQ_API_KEY environment variable."
+
+    prompt = (
+        f"You are a senior full‑stack developer. Based on the following description, "
+        f"write a complete, runnable {language} web application. Include all necessary "
+        "files (HTML, CSS, JavaScript, or a single script) and comment the code. "
+        "Return the code only, without extra explanation.\n\n"
+        f"Description:\n{idea}"
+    )
+
+    try:
+        response = groq_client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.3,
+            max_tokens=4000,
+        )
+        generated = response.choices[0].message.content
+        return generated.strip() if generated else "No content returned from Groq."
+    except Exception as e:
+        return f"Error communicating with Groq: {e}"
+
+# ----------------------------------------------------------------------
+# Gradio UI
 # ----------------------------------------------------------------------
 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,
+    gr.Markdown("# Code Improver & Web Generator")
+
+    with gr.Tabs():
+        # --------------------------------------------------------------
+        # Tab 1: Improve existing code
+        # --------------------------------------------------------------
+        with gr.TabItem("Improve Code"):
+            with gr.Row():
+                with gr.Column():
+                    lang_select_improve = gr.Dropdown(
+                        choices=SUPPORTED_LANGUAGES,
+                        label="Language",
+                        value="Python",
+                    )
+                    code_input = gr.Textbox(
+                        label="Original Code",
+                        placeholder="Paste your code here",
+                        lines=15,
+                    )
+                with gr.Column():
+                    improved_output = gr.Textbox(
+                        label="Improved Code",
+                        placeholder="Improved code will appear here",
+                        lines=15,
+                    )
+            improve_btn = gr.Button("Improve")
+            improve_btn.click(
+                fn=improve_code,
+                inputs=[code_input, lang_select_improve],
+                outputs=improved_output,
             )
-        with gr.Column():
-            code_output = gr.Textbox(
-                label="Improved Code",
-                placeholder="Improved code will appear here",
-                lines=20,
-                max_lines=200,
+
+        # --------------------------------------------------------------
+        # Tab 2: Generate web project from idea
+        # --------------------------------------------------------------
+        with gr.TabItem("Generate Web"):
+            with gr.Row():
+                with gr.Column():
+                    lang_select_generate = gr.Dropdown(
+                        choices=SUPPORTED_LANGUAGES,
+                        label="Target Language",
+                        value="HTML",
+                    )
+                    idea_input = gr.Textbox(
+                        label="Web Idea",
+                        placeholder="Describe the web page or app you want",
+                        lines=12,
+                    )
+                with gr.Column():
+                    generated_output = gr.Textbox(
+                        label="Generated Code",
+                        placeholder="The generated code will appear here",
+                        lines=12,
+                    )
+            generate_btn = gr.Button("Generate")
+            generate_btn.click(
+                fn=generate_web_code,
+                inputs=[idea_input, lang_select_generate],
+                outputs=generated_output,
             )
-    improve_btn = gr.Button("Improve")
-    improve_btn.click(fn=improve_code, inputs=code_input, outputs=code_output)
 
 demo.launch()