Update app.py

app.py

@@ -391,6 +391,55 @@ model = AutoModelForCausalLM.from_pretrained(
 
 import gradio as gr
 import torch
+def generate_from_prompt(prompt):
+    try:
+        messages = [{"role": "user", "content": prompt}]
+        input_ids = tokenizer.apply_chat_template(
+            messages,
+            tokenize=True,
+            return_tensors="pt",
+            add_generation_prompt=True,
+        ).to(model.device)
+
+        with torch.no_grad():
+            outputs = model.generate(
+                input_ids,
+                max_new_tokens=4096,
+                pad_token_id=tokenizer.eos_token_id,
+                eos_token_id=tokenizer.eos_token_id,
+                do_sample=True,
+                temperature=0.7,
+                top_p=0.95,
+            )
+
+        response = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
+        return response.strip()
+
+    except Exception as e:
+        return f"Error during generation: {e}"
+def generate_model_from_description(description):
+    if not description.strip():
+        return "Error: Description is required."
+
+    prompt = f"""
+You are an expert C# developer.
+
+Generate a clean, professional C# entity model class based on the following project description:
+
+{description}
+
+Please output only the C# class code.
+"""
+
+    code = generate_from_prompt(prompt)
+
+    # تنظيف الكود من علامات البرمجة اذا موجودة
+    if code.startswith("```csharp"):
+        code = code[len("```csharp"):].strip()
+    if code.endswith("```"):
+        code = code[:-len("```")].strip()
+
+    return code
 
 # افتراضياً لديك هذه الدالة كما كتبتها
 @spaces.GPU
@@ -491,9 +540,27 @@ with gr.Blocks(title="C# Validator Generator with 8 Tabs") as demo:
                 inputs=[model_names_input, shared_model_structure_input, shared_template_input],
                 outputs=[batch_code_output],
             )
+        with gr.Tab("Generate Model from Description"):
+            with gr.Row():
+                with gr.Column():
+                    description_input = gr.Textbox(
+                        label="Project Description",
+                        lines=10,
+                        placeholder="Enter your project or model description here..."
+                    )
+                    generate_model_desc_btn = gr.Button("Generate Model Code")
+        
+                with gr.Column():
+                    model_code_output_desc = gr.Code(label="Generated Model Code", lines=25)
+        
+            generate_model_desc_btn.click(
+                fn=generate_model_from_description,
+                inputs=[description_input],
+                outputs=[model_code_output_desc]
+            )
 
         # Tab 3 to Tab 8 placeholders for future features
-        for i in range(3, 9):
+        for i in range(4, 9):
             with gr.Tab(f"Tab {i} - Placeholder"):
                 gr.Markdown(f"Content for Tab {i} goes here...")