Update app.py

app.py

@@ -37,10 +37,27 @@ def validate_inputs(gen_prompt, max_tokens, temperature, seed):
 
 def generate_code(gen_prompt, max_tokens, temperature=0.6, seed=42):
     validate_inputs(gen_prompt, max_tokens, temperature, seed)
-    
-    # Rest of code generation logic here
-    
-    return generate_code
+
+    # Encode the input prompt
+    input_ids = tokenizer.encode(gen_prompt, return_tensors="pt")
+
+    # Set seed for reproducibility
+    set_seed(seed)
+
+    # Generate code tokens
+    output = model.generate(
+        input_ids,
+        max_length=max_tokens + input_ids.shape[-1],
+        temperature=temperature,
+        pad_token_id=tokenizer.eos_token_id,
+        num_return_sequences=1
+    )
+
+    # Decode the generated tokens into Python code
+    generated_code = tokenizer.decode(output[:, input_ids.shape[-1]:][0], skip_special_tokens=True)
+
+    return generated_code
+
 
 
 def save_to_text_file(output_text):