Update app.py

app.py

@@ -53,7 +53,6 @@ def generate_response(messages, max_length=512, temperature=0.7, top_p=0.9):
         print(f"Error during generation: {e}")
         return f"An error occurred: {str(e)}", num_generated_tokens
 
-# --- Input Processing Function (Takes single system prompt) ---
 @spaces.GPU # Keep ZeroGPU decorator
 def process_input(
     analysis_mode, # Mode selector
@@ -66,19 +65,18 @@ def process_input(
     """Process inputs based on selected analysis mode using the provided system prompt."""
     print(f"GPU requested via decorator, starting processing in mode: {analysis_mode}")
 
-    # Construct user content based on mode
-    if analysis_mode == "Frequency Only":
-        user_content = f"Player Move Frequency Stats (Long-Term):\n{player_stats}"
-    elif analysis_mode == "Markov Prediction Only":
-        user_content = "Analyze based on the system prompt instructions."
-    else:
-        return "Invalid analysis mode selected.", "", "0 seconds", 0
-
-    # Create the messages list using the system_prompt from the UI
+    # Create the messages list using the system_prompt from the UI directly
     messages = []
     if system_prompt and system_prompt.strip():
         messages.append({"role": "system", "content": system_prompt})
-    messages.append({"role": "user", "content": user_content})
+    
+    # Add an empty user message or specific content based on analysis mode
+    if analysis_mode == "Frequency Only":
+        user_content = f"Player Move Frequency Stats (Long-Term):\n{player_stats}"
+        messages.append({"role": "user", "content": user_content})
+    else:  # For Markov Prediction or any other mode
+        # Just add an empty user message to trigger the model response
+        messages.append({"role": "user", "content": ""})
 
     # --- Time Measurement Start ---
     start_time = time.time()
@@ -95,8 +93,11 @@ def process_input(
     end_time = time.time()
     duration = round(end_time - start_time, 2)
 
-    # For display purposes
-    display_prompt = f"Selected Mode: {analysis_mode}\nSystem Prompt:\n{system_prompt}\n\n------\n\nUser Content:\n{user_content}"
+    # For display purposes - show what was actually sent to the model
+    if analysis_mode == "Frequency Only":
+        display_prompt = f"Selected Mode: {analysis_mode}\nSystem Prompt:\n{system_prompt}\n\n------\n\nUser Content:\n{user_content}"
+    else:
+        display_prompt = f"Selected Mode: {analysis_mode}\nSystem Prompt:\n{system_prompt}\n\n------\n\nUser Content:\n[Empty]"
 
     print(f"Processing finished in {duration} seconds.")
     # Return all results including time and tokens