Update app.py

app.py

@@ -1,19 +1,23 @@
 import gradio as gr
 from transformers import pipeline
 
-# Initialize the text generation pipeline
+# Initialize the text generation pipelines
 pipe = pipeline("text-generation", model="akhaliq/MyGemmaGradioCoder")
+pipe2 = pipeline("text-generation", model="google/gemma-3-270m-it")
 
-def generate_code(user_input):
+def generate_code(user_input, model_choice="Model 1"):
     """
-    Generate code based on user input using the Gemma model
+    Generate code based on user input using the selected Gemma model
     """
     messages = [
         {"role": "user", "content": user_input},
     ]
     
+    # Select pipeline based on model choice
+    selected_pipe = pipe if model_choice == "Model 1 (MyGemmaGradioCoder)" else pipe2
+    
     # Generate response from the model
-    response = pipe(messages, max_new_tokens=512, temperature=0.7, do_sample=True)
+    response = selected_pipe(messages, max_new_tokens=512, temperature=0.7, do_sample=True)
     
     # Extract the generated text from the response
     generated_text = response[0]['generated_text']
@@ -32,14 +36,25 @@ def generate_code(user_input):
             return generated_text.split(user_input)[-1].strip()
         return generated_text
 
+def generate_both(user_input):
+    """
+    Generate code from both models for comparison
+    """
+    output1 = generate_code(user_input, "Model 1 (MyGemmaGradioCoder)")
+    output2 = generate_code(user_input, "Model 2 (gemma-3-270m-it)")
+    return output1, output2
+
 # Create the Gradio interface
-with gr.Blocks(title="Text to Code Generator", theme=gr.themes.Soft()) as demo:
+with gr.Blocks(title="Text to Code Generator - Model Comparison", theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
-        # 🚀 Text to Code Generator
+        # 🚀 Text to Code Generator - Model Comparison
+        
+        Compare code generation from two different Gemma models:
+        - **Model 1**: akhaliq/MyGemmaGradioCoder
+        - **Model 2**: google/gemma-3-270m-it
         
-        Generate code from natural language descriptions using the Gemma Gradio Coder model.
-        Simply describe what you want to build, and the AI will generate the corresponding code!
+        Simply describe what you want to build, and see how each model responds!
         """
     )
     
@@ -54,7 +69,7 @@ with gr.Blocks(title="Text to Code Generator", theme=gr.themes.Soft()) as demo:
             )
             
             with gr.Row():
-                generate_btn = gr.Button("Generate Code", variant="primary", scale=2)
+                generate_btn = gr.Button("Generate from Both Models", variant="primary", scale=2)
                 clear_btn = gr.ClearButton([input_text], value="Clear", scale=1)
             
             # Examples section
@@ -70,44 +85,71 @@ with gr.Blocks(title="Text to Code Generator", theme=gr.themes.Soft()) as demo:
                 label="Example Prompts"
             )
         
-        with gr.Column(scale=1):
-            # Output section
-            output_code = gr.Code(
-                label="Generated Code",
-                language="python",
-                lines=20,
-                interactive=True,
-                show_line_numbers=True,
-                wrap_lines=True,
-                autocomplete=True
-            )
-            
+        with gr.Column(scale=2):
+            # Output section - Two columns for comparison
             with gr.Row():
-                copy_btn = gr.Button("📋 Copy Code", scale=1)
+                with gr.Column():
+                    gr.Markdown("### Model 1: MyGemmaGradioCoder")
+                    output_code1 = gr.Code(
+                        label="Generated Code (Model 1)",
+                        language="python",
+                        lines=15,
+                        interactive=True,
+                        show_line_numbers=True,
+                        wrap_lines=True,
+                        autocomplete=True
+                    )
+                    copy_btn1 = gr.Button("📋 Copy Code", scale=1)
+                
+                with gr.Column():
+                    gr.Markdown("### Model 2: gemma-3-270m-it")
+                    output_code2 = gr.Code(
+                        label="Generated Code (Model 2)",
+                        language="python",
+                        lines=15,
+                        interactive=True,
+                        show_line_numbers=True,
+                        wrap_lines=True,
+                        autocomplete=True
+                    )
+                    copy_btn2 = gr.Button("📋 Copy Code", scale=1)
             
     # Add event handlers
     generate_btn.click(
-        fn=generate_code,
+        fn=generate_both,
         inputs=input_text,
-        outputs=output_code,
+        outputs=[output_code1, output_code2],
         api_name="generate"
     )
     
     input_text.submit(
-        fn=generate_code,
+        fn=generate_both,
         inputs=input_text,
-        outputs=output_code
+        outputs=[output_code1, output_code2]
+    )
+    
+    # Copy functionality for both outputs
+    copy_btn1.click(
+        fn=None,
+        inputs=output_code1,
+        outputs=None,
+        js="""
+        (code) => {
+            navigator.clipboard.writeText(code);
+            alert('Code from Model 1 copied to clipboard!');
+            return null;
+        }
+        """
     )
     
-    # Copy functionality (using JavaScript)
-    copy_btn.click(
+    copy_btn2.click(
         fn=None,
-        inputs=output_code,
+        inputs=output_code2,
         outputs=None,
         js="""
         (code) => {
             navigator.clipboard.writeText(code);
-            alert('Code copied to clipboard!');
+            alert('Code from Model 2 copied to clipboard!');
             return null;
         }
         """
@@ -122,7 +164,9 @@ with gr.Blocks(title="Text to Code Generator", theme=gr.themes.Soft()) as demo:
         - Include details about inputs, outputs, and edge cases
         - You can edit the generated code directly in the output box
         
-        Powered by [akhaliq/MyGemmaGradioCoder](https://huggingface.co/akhaliq/MyGemmaGradioCoder)
+        **Models:**
+        - [akhaliq/MyGemmaGradioCoder](https://huggingface.co/akhaliq/MyGemmaGradioCoder)
+        - [google/gemma-3-270m-it](https://huggingface.co/google/gemma-3-270m-it)
         """
     )