Update app.py

app.py

@@ -1,6 +1,5 @@
-# AI Persona Simulator - Final Version
+# AI Persona Simulator - Stateless GPU Optimized Version
 import gradio as gr
-import transformers
 import torch
 from transformers import pipeline
 from duckduckgo_search import DDGS
@@ -21,7 +20,7 @@ logging.basicConfig(
 MODEL_ID = "google/gemma-3-1b-it"
 MAX_GPU_MEMORY = "60GiB"
 
-# --- Model Loading (GPU Isolated) ---
+# --- GPU-Isolated Functions ---
 @GPU(memory=60)
 def load_model():
     """Load the Gemma 3 1B model without quantization for full precision."""
@@ -40,9 +39,9 @@ def load_model():
         print(f"FATAL Error loading model '{MODEL_ID}': {e}")
         raise e
 
-# --- LLM-Based Request Validation ---
-def check_request_with_llm(name, context):
-    """Use LLM to check if request is appropriate before processing"""
+@GPU(memory=60)
+def validate_request(name, context):
+    """LLM-based request validation using isolated GPU function"""
     try:
         pipe = load_model()
         
@@ -264,8 +263,7 @@ Additional context for the simulation: {context}
 Maintain this persona consistently. Respond naturally based on the profile. Do not mention that you are an AI or a simulation. If asked about details not in the profile, you can be evasive or state you don't know/remember, consistent with the persona."""
     
     prompt = [
-        {"role": "system", "content": """You are an expert AI prompt engineer specializing in character simulation. Create a concise system prompt that instructs the LLM to embody the character based on the profile. The prompt must: 1. Define core personality and speaking style. 2. Specify how to handle unknown topics. 3. Prohibit breaking character or mentioning AI nature. Output ONLY the system prompt itself."""
-        },
+        {"role": "system", "content": """You are an expert AI prompt engineer specializing in character simulation. Create a concise system prompt that instructs the LLM to embody the character based on the profile. The prompt must: 1. Define core personality and speaking style. 2. Specify how to handle unknown topics. 3. Prohibit breaking character or mentioning AI nature. Output ONLY the system prompt itself."""},
         {"role": "user", "content": f"""Create a system prompt for an AI to simulate the character '{name}'. Context for simulation: {context} Character Profile:
 {enhanced_profile}
 Generate the system prompt based *only* on the profile and context provided."""}
@@ -340,7 +338,7 @@ class PersonaChat:
         """Orchestrates persona creation: validation, search, enhance, generate prompt."""
         try:
             # First validate the request with LLM
-            is_valid = check_request_with_llm(name, context)
+            is_valid = validate_request(name, context)
             if not is_valid:
                 warning = "This request has been flagged as inappropriate. We cannot create personas that involve minors, vulnerable individuals, or potentially harmful scenarios."
                 yield warning, "", "", [{"role": "system", "content": warning}]