Update assistant.py

assistant.py

@@ -2,19 +2,38 @@ from command_classifier import classify_command
 from executor import execute_command
 
 def get_assistant_response(pipe, command: str, execute: bool) -> str:
+    """
+    Process the user command using the language model pipeline and optionally execute a system command.
+
+    Args:
+        pipe: The Hugging Face model pipeline
+        command (str): The user's text input
+        execute (bool): Whether or not to execute a real command on the machine
+
+    Returns:
+        str: Response from the assistant (and optionally OS output)
+    """
     command_type = classify_command(command)
+
+    # If execute flag is enabled, try running a system command
     if execute:
-        result = execute_command(command_type, command)
-        if result:
-            return result
+        exec_output = execute_command(command_type, command)
+        if exec_output:
+            return f"🛠️ Executed:\n{exec_output}"
 
+    # Format prompt for Qwen2.5-VL instruction-tuned model
     prompt = f"<|user|>\n{command}\n<|assistant|>\n"
+
     try:
         output = pipe(prompt)[0]["generated_text"]
+
+        # Cleanly extract assistant response
         if "<|assistant|>" in output:
             response = output.split("<|assistant|>")[-1].strip()
         else:
             response = output.strip()
+
         return f"🤖 ({command_type}) {response}"
+
     except Exception as e:
-        return f"⚠️ Error: {e}"
+        return f"⚠️ Model error: {e}"