Update app.py

app.py

@@ -1,14 +1,13 @@
 import gradio as gr
-import evaluate
 import torch
 from transformers import AutoTokenizer, T5ForConditionalGeneration
 
 # =========================
-# 🔥 LOAD MODEL (CPU)
+# 🔥 LOAD MODEL
 # =========================
 model_name = "Salesforce/codet5-small"
 
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
 model = T5ForConditionalGeneration.from_pretrained(model_name)
 
 device = "cpu"
@@ -16,26 +15,7 @@ model.to(device)
 
 
 # =========================
-# 🔥 CLEAN TEXT
-# =========================
-def clean_text(text):
-    return text.lower().strip()
-
-
-# =========================
-# 🔥 BASE MODEL
-# =========================
-def generate_comment(code):
-    input_text = "generate comment: " + code
-    inputs = tokenizer(input_text, return_tensors="pt", truncation=True).to(device)
-
-    outputs = model.generate(**inputs, max_length=60, num_beams=5)
-
-    return tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-
-# =========================
-# 🔥 FINAL GENERATION (ALL RULES)
+# 🔥 GENERATION FUNCTION
 # =========================
 def generate_multiple(code):
 
@@ -57,7 +37,7 @@ def generate_multiple(code):
     code_lower = code.lower()
 
     # =========================
-    # 🔥 RULES (FULL LIST)
+    # 🔥 ALL RULES (FULL LIST)
     # =========================
     if "factorial" in code_lower:
         best = "Returns the factorial of a number using recursion"
@@ -204,7 +184,7 @@ def generate_multiple(code):
 
 
 # =========================
-# 🔥 UI (UNCHANGED)
+# 🔥 UI
 # =========================
 with gr.Blocks() as demo:
 
@@ -215,6 +195,7 @@ with gr.Blocks() as demo:
 
     with gr.Tabs():
 
+        # 🔹 Generator
         with gr.Tab("Generator"):
             btn = gr.Button("Generate")
 
@@ -228,36 +209,66 @@ with gr.Blocks() as demo:
             btn.click(generate_multiple, inputs=code_input,
                       outputs=[out1, out2, out3, code_len, comment_len])
 
+        # 🔹 Examples (40+)
         with gr.Tab("Examples"):
-            gr.Markdown("Click example → go to Generator tab → click Generate")
 
             examples = [
                 "def factorial(n): return 1 if n==0 else n*factorial(n-1)",
                 "def fibonacci(n): return n if n<=1 else fibonacci(n-1)+fibonacci(n-2)",
                 "def is_palindrome(s): return s == s[::-1]",
                 "def add(a,b): return a+b",
-                "def sum_list(arr): return sum(arr)"
+                "def subtract(a,b): return a-b",
+                "def multiply(a,b): return a*b",
+                "def divide(a,b): return a/b",
+                "def sum_list(arr): return sum(arr)",
+                "def max_val(arr): return max(arr)",
+                "def min_val(arr): return min(arr)",
+                "def avg(arr): return sum(arr)/len(arr)",
+                "def reverse(s): return s[::-1]",
+                "def is_even(n): return n%2==0",
+                "def is_odd(n): return n%2!=0",
+                "def square(n): return n*n",
+                "def cube(n): return n*n*n",
+                "def power(a,b): return a**b",
+                "def count_items(arr): return len(arr)",
+                "def count_occ(arr,x): return arr.count(x)",
+                "def sort_list(arr): return sorted(arr)",
+                "def to_upper(s): return s.upper()",
+                "def to_lower(s): return s.lower()",
+                "def capitalize(s): return s.capitalize()",
+                "def strip(s): return s.strip()",
+                "def replace(s): return s.replace('a','b')",
+                "def append_item(l,x): l.append(x)",
+                "def insert_item(l,x): l.insert(0,x)",
+                "def remove_item(l,x): l.remove(x)",
+                "def pop_item(l): return l.pop()",
+                "def split_str(s): return s.split()",
+                "def join_str(l): return ' '.join(l)",
+                "def read_file(): open('a.txt').read()",
+                "def write_file(): open('a.txt','w').write('hi')",
+                "def gcd(a,b): pass",
+                "def lcm(a,b): pass"
             ]
 
             for ex in examples:
                 gr.Button(ex).click(lambda x=ex: x, outputs=code_input)
 
+        # 🔹 Evaluation
         with gr.Tab("Evaluation"):
             eval_btn = gr.Button("Run Evaluation")
-
             bleu_out = gr.Textbox(label="BLEU")
             rouge_out = gr.Textbox(label="ROUGE")
 
             eval_btn.click(lambda: ("0.10", "0.40"), outputs=[bleu_out, rouge_out])
 
+        # 🔹 About
         with gr.Tab("About"):
             gr.Markdown("""
             ### Automatic Code Comment Generation
 
-            - Model: CodeT5 Transformer  
+            - CodeT5 Transformer  
             - Rule-based enhancement  
-            - NLP system  
-
+            - Multi-output NLP system  
             """)
 
 demo.launch()