Update app.py

app.py

@@ -18,7 +18,6 @@ class NordTheme(Base):
             font_mono=(fonts.GoogleFont("Fira Code"), "ui-monospace", "monospace"),
         )
         self.set(
-            # Color Palette
             body_background_fill="#2E3440",
             body_text_color="#ECEFF4",
             block_background_fill="#3B4252",
@@ -26,37 +25,27 @@ class NordTheme(Base):
             block_border_color="#4C566A",
             block_label_background_fill="#434C5E",
             block_label_text_color="#ECEFF4",
-            
             input_background_fill="#434C5E",
             input_border_color="transparent",
-            input_text_color="#ECEFF4",
-            
-            # Button Styles
             button_primary_background_fill="#5E81AC",
             button_primary_background_fill_hover="#81A1C1",
             button_primary_text_color="#ECEFF4",
             button_secondary_background_fill="#4C566A",
             button_secondary_background_fill_hover="#5a657a",
             button_secondary_text_color="#ECEFF4",
-            
-            # Other Elements
             border_color_accent="#5E81AC",
             background_fill_primary_dark="#2E3440",
             color_accent_soft="#4c566a",
-            
-            # Sizing
-            radius_sm=sizes.radius_sm,
-            radius_md=sizes.radius_md,
-            radius_lg=sizes.radius_lg,
             block_radius="12px",
             button_radius="8px",
         )
 
 # --- UI Configuration ---
 
+# UPDATED: "Natural Language" is added to the list.
 LANGUAGES = [
-    'Python', 'JavaScript', 'TypeScript', 'Java', 'C++', 'C#', 'C', 'Go', 
-    'Rust', 'Swift', 'Kotlin', 'PHP', 'Ruby', 'Scala', 'R', 'MATLAB', 
+    'Natural Language', 'Python', 'JavaScript', 'TypeScript', 'Java', 'C++', 'C#', 
+    'C', 'Go', 'Rust', 'Swift', 'Kotlin', 'PHP', 'Ruby', 'Scala', 'R', 'MATLAB', 
     'Perl', 'Haskell', 'Lua', 'Dart', 'Elixir', 'F#', 'Clojure', 'SQL'
 ]
 
@@ -83,22 +72,52 @@ print(fibonacci(10))"""
 def convert_code(
     source_code: str, source_lang: str, target_lang: str, model: str, api_key: str
 ):
+    """
+    Handles three conversion scenarios:
+    1. Natural Language -> Code
+    2. Code -> Natural Language
+    3. Code -> Code
+    """
     if not source_code.strip():
-        raise gr.Error("Please enter some code to convert.")
+        raise gr.Error("Please enter some code or a description to convert.")
     if not api_key.strip():
         raise gr.Error("OpenRouter API key is required.")
     if not model or not model.strip():
         raise gr.Error("Please select or enter a model for the conversion.")
+    
+    # Handle the edge case where both are "Natural Language"
+    if source_lang == "Natural Language" and target_lang == "Natural Language":
+        raise gr.Error("Please select a programming language for either the source or the target.")
 
     client = openai.OpenAI(base_url="https://openrouter.ai/api/v1", api_key=api_key)
-    prompt = (
-        f"You are an expert programmer. Convert the following {source_lang} code "
-        f"to {target_lang}. Your response should only contain the raw, converted code. "
-        "Do not include any explanations, markdown formatting (like ```), or extra text."
-        f"\n\n--- Start of {source_lang} Code ---\n"
-        f"{source_code}\n"
-        f"--- End of {source_lang} Code ---"
-    )
+
+    # UPDATED: Logic to generate a prompt based on the conversion type
+    if source_lang == "Natural Language":
+        # Scenario 1: Natural Language -> Code
+        prompt = (
+            f"You are a programming expert. Based on the following description, write a complete and functional code snippet in {target_lang}. "
+            f"The user's request is: '{source_code}'.\n\n"
+            f"Your response must only contain the raw {target_lang} code without any explanations, comments, or markdown formatting (like ```)."
+        )
+    elif target_lang == "Natural Language":
+        # Scenario 2: Code -> Natural Language
+        prompt = (
+            f"You are a programming expert. Explain the following {source_lang} code in simple, natural English. "
+            f"Describe what the code does, its main logic, its inputs, and what it outputs. Do not include any code in your explanation.\n\n"
+            f"--- Start of {source_lang} Code ---\n"
+            f"{source_code}\n"
+            f"--- End of {source_lang} Code ---"
+        )
+    else:
+        # Scenario 3: Code -> Code (original logic)
+        prompt = (
+            f"You are an expert programmer. Convert the following {source_lang} code to {target_lang}. "
+            "Your response must only contain the raw, converted code. Do not include any explanations, comments, or markdown formatting (like ```)."
+            f"\n\n--- Start of {source_lang} Code ---\n"
+            f"{source_code}\n"
+            f"--- End of {source_lang} Code ---"
+        )
+    
     try:
         completion = client.chat.completions.create(
             model=model,
@@ -113,49 +132,43 @@ def convert_code(
 
 # --- Gradio User Interface ---
 
+# Helper function to update Code component's language
+def update_code_language(lang: str):
+    if lang == "Natural Language":
+        return gr.update(language="text", placeholder="Describe the code you want here...")
+    return gr.update(language=lang.lower(), placeholder=f"Enter your {lang} code here...")
+
 with gr.Blocks(theme=NordTheme()) as app:
     # Header
     gr.HTML(
         """
         <div style="display: flex; align-items: center; gap: 12px; padding: 10px;">
-            <svg xmlns="[http://www.w3.org/2000/svg](http://www.w3.org/2000/svg)" width="32" height="32" viewBox="0 0 24 24" fill="none" stroke="#81A1C1" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="m18 16 4-4-4-4"/><path d="m6 8-4 4 4 4"/><path d="m14.5 4-5 16"/></svg>
+            <svg xmlns="http://www.w3.org/2000/svg" width="32" height="32" viewBox="0 0 24 24" fill="none" stroke="#81A1C1" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"><path d="m18 16 4-4-4-4"/><path d="m6 8-4 4 4 4"/><path d="m14.5 4-5 16"/></svg>
             <div>
                 <h1 style="font-size: 1.5rem; font-weight: 700; color: #ECEFF4; margin: 0;">CodeVerter</h1>
-                <p style="font-size: 0.875rem; color: #D8DEE9; margin: 0;">Programming Language Converter via OpenRouter</p>
+                <p style="font-size: 0.875rem; color: #D8DEE9; margin: 0;">Code & Natural Language Converter</p>
             </div>
         </div>
         """
     )
 
-    # Configuration Accordion
     with gr.Accordion("⚙️ Configuration", open=True):
-        openrouter_key = gr.Textbox(
-            label="OpenRouter API Key",
-            placeholder="Enter your OpenRouter API key...",
-            type="password",
-        )
-        model_selection = gr.Dropdown(
-            label="Select or Enter a Model Name",
-            choices=MODELS,
-            value=MODELS[0] if MODELS else None,
-            allow_custom_value=True,
-        )
+        openrouter_key = gr.Textbox(label="OpenRouter API Key", placeholder="Enter your OpenRouter API key...", type="password")
+        model_selection = gr.Dropdown(label="Select or Enter a Model Name", choices=MODELS, value=MODELS[0] if MODELS else None, allow_custom_value=True)
 
-    # Main conversion interface
     with gr.Row(equal_height=False):
         with gr.Column(scale=1):
             source_lang_selection = gr.Dropdown(label="Source Language", choices=LANGUAGES, value="Python")
-            source_code_input = gr.Code(label="Source Code", language="python", value=DEFAULT_SOURCE_CODE, lines=15)
+            source_code_input = gr.Code(label="Source", language="python", value=DEFAULT_SOURCE_CODE, lines=15)
         with gr.Column(scale=1):
             target_lang_selection = gr.Dropdown(label="Target Language", choices=LANGUAGES, value="JavaScript")
-            target_code_output = gr.Code(label="Converted Code", language="javascript", lines=15, interactive=False)
+            target_code_output = gr.Code(label="Result", language="javascript", lines=15, interactive=False)
     
-    # Sync language highlighting with dropdowns
-    source_lang_selection.change(lambda x: gr.update(language=x.lower()), inputs=source_lang_selection, outputs=source_code_input)
-    target_lang_selection.change(lambda x: gr.update(language=x.lower()), inputs=target_lang_selection, outputs=target_code_output)
+    # UPDATED: Use a helper function for cleaner UI logic
+    source_lang_selection.change(fn=update_code_language, inputs=source_lang_selection, outputs=source_code_input)
+    target_lang_selection.change(lambda lang: gr.update(language=lang.lower() if lang != "Natural Language" else "text"), inputs=target_lang_selection, outputs=target_code_output)
 
-    # Convert button and action
-    convert_button = gr.Button("Convert Code", variant="primary", scale=1)
+    convert_button = gr.Button("Convert", variant="primary", scale=1)
     convert_button.click(
         fn=convert_code,
         inputs=[source_code_input, source_lang_selection, target_lang_selection, model_selection, openrouter_key],
@@ -163,10 +176,9 @@ with gr.Blocks(theme=NordTheme()) as app:
         api_name="convert"
     )
 
-    # Footer
     gr.HTML(
         """<div style="text-align: center; margin-top: 24px; color: #a0aec0; font-size: 0.875rem;">
-        <p>CodeVerter uses LLMs via OpenRouter. Results may require manual review.</p>
+        <p>Powered by OpenRouter. Results may require manual review.</p>
         </div>"""
     )